nsfwalex commited on
Commit
3797680
·
verified ·
1 Parent(s): b176ecd

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. inference_manager.py +4 -1
inference_manager.py CHANGED
@@ -252,7 +252,10 @@ class InferenceManager:
252
  samplers = {
253
  "Euler a": EulerAncestralDiscreteScheduler.from_config(temp_pipeline.scheduler.config),
254
  "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(temp_pipeline.scheduler.config, use_karras_sigmas=True),
255
- "DPM2 a": DPMSolverMultistepScheduler.from_config(temp_pipeline.scheduler.config)
 
 
 
256
  }
257
 
258
  # Set the scheduler based on the selected sampler
 
252
  samplers = {
253
  "Euler a": EulerAncestralDiscreteScheduler.from_config(temp_pipeline.scheduler.config),
254
  "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(temp_pipeline.scheduler.config, use_karras_sigmas=True),
255
+ "DPM2 a": DPMSolverMultistepScheduler.from_config(temp_pipeline.scheduler.config),
256
+ "DPM++ SDE": DPMSolverSDEScheduler.from_config(temp_pipeline.scheduler.config),
257
+ "DPM++ 2M SDE": DPMSolverSDEScheduler.from_config(temp_pipeline.scheduler.config, use_2m=True),
258
+ "DPM++ 2S a": DPMSolverMultistepScheduler.from_config(temp_pipeline.scheduler.config, use_2s=True)
259
  }
260
 
261
  # Set the scheduler based on the selected sampler