ford442 commited on
Commit
e442a24
·
1 Parent(s): cf49fe3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def load_and_prepare_model(model_id):
120
  vae=vae,
121
  )
122
  #sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2",use_karras_sigmas=True, algorithm_type="dpmsolver++")
123
- sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2",use_karras_sigmas=True, algorithm_type="dpmsolver++")
124
 
125
  pipe.scheduler=sched
126
  #pipe.to(device=device, dtype=torch.bfloat16)
 
120
  vae=vae,
121
  )
122
  #sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2",use_karras_sigmas=True, algorithm_type="dpmsolver++")
123
+ sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2", algorithm_type="dpmsolver++")
124
 
125
  pipe.scheduler=sched
126
  #pipe.to(device=device, dtype=torch.bfloat16)