ford442 commited on
Commit
d2ce2d8
·
1 Parent(s): eec6afd

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
  use_safetensors=True,
121
  vae=vae,
122
  )
123
- sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear",use_karras_sigmas=True, algorithm_type="dpmsolver++")
124
  #sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="linear", algorithm_type="dpmsolver++")
125
  #sched = DDIMScheduler.from_config(pipe.scheduler.config)
126
  pipe.scheduler=sched
 
120
  use_safetensors=True,
121
  vae=vae,
122
  )
123
+ sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", algorithm_type="dpmsolver++")
124
  #sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="linear", algorithm_type="dpmsolver++")
125
  #sched = DDIMScheduler.from_config(pipe.scheduler.config)
126
  pipe.scheduler=sched