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")
|