patrickvonplaten commited on
Commit
a1fd59a
·
1 Parent(s): fcc7a59
Files changed (1) hide show
  1. 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")