Commit
·
a1fd59a
1
Parent(s):
fcc7a59
up
Browse files- run_xl_ediffi.py +0 -14
run_xl_ediffi.py
CHANGED
|
@@ -37,20 +37,6 @@ pipe_high_noise = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.
|
|
| 37 |
pipe_high_noise.scheduler = scheduler
|
| 38 |
pipe_high_noise.to("cuda")
|
| 39 |
|
| 40 |
-
scheduler = DPMSolverMultistepScheduler(
|
| 41 |
-
beta_start=0.00085,
|
| 42 |
-
beta_end=0.012,
|
| 43 |
-
beta_schedule="scaled_linear",
|
| 44 |
-
prediction_type="epsilon",
|
| 45 |
-
num_train_timesteps=1000,
|
| 46 |
-
trained_betas=None,
|
| 47 |
-
thresholding=False,
|
| 48 |
-
algorithm_type="dpmsolver++",
|
| 49 |
-
solver_type="midpoint",
|
| 50 |
-
lower_order_final=True,
|
| 51 |
-
use_karras_sigmas=True,
|
| 52 |
-
)
|
| 53 |
-
|
| 54 |
pipe_low_noise = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
|
| 55 |
pipe_low_noise.scheduler = scheduler
|
| 56 |
pipe_low_noise.to("cuda")
|
|
|
|
| 37 |
pipe_high_noise.scheduler = scheduler
|
| 38 |
pipe_high_noise.to("cuda")
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
pipe_low_noise = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
|
| 41 |
pipe_low_noise.scheduler = scheduler
|
| 42 |
pipe_low_noise.to("cuda")
|