gokaygokay commited on
Commit
38bed11
1 Parent(s): 7452dff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -93,6 +93,7 @@ samplers = {
93
  "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(pipe_pony.scheduler.config, use_karras_sigmas=True),
94
  "Heun": HeunDiscreteScheduler.from_config(pipe_pony.scheduler.config),
95
  # New samplers
 
96
  "DPM++ 2M": DPMSolverMultistepScheduler.from_config(pipe_pony.scheduler.config),
97
  "DDIM": DDIMScheduler.from_config(pipe_pony.scheduler.config),
98
  "LMS": LMSDiscreteScheduler.from_config(pipe_pony.scheduler.config),
 
93
  "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(pipe_pony.scheduler.config, use_karras_sigmas=True),
94
  "Heun": HeunDiscreteScheduler.from_config(pipe_pony.scheduler.config),
95
  # New samplers
96
+ "DPM++ 2M SDE Karras": DPMSolverMultistepScheduler.from_config(pipe_pony.scheduler.config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++")
97
  "DPM++ 2M": DPMSolverMultistepScheduler.from_config(pipe_pony.scheduler.config),
98
  "DDIM": DDIMScheduler.from_config(pipe_pony.scheduler.config),
99
  "LMS": LMSDiscreteScheduler.from_config(pipe_pony.scheduler.config),