Update inference_manager.py
Browse files- 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
|