change realistic to DPM++ 2M SDE Karras
Browse files- handler.py +5 -0
handler.py
CHANGED
|
@@ -242,8 +242,13 @@ class EndpointHandler:
|
|
| 242 |
# https://stablediffusionapi.com/docs/a1111schedulers/
|
| 243 |
|
| 244 |
# DPM++ 2M Karras
|
|
|
|
|
|
|
|
|
|
|
|
|
| 245 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
| 246 |
pipe.scheduler.config,
|
|
|
|
| 247 |
use_karras_sigmas=True,
|
| 248 |
)
|
| 249 |
|
|
|
|
| 242 |
# https://stablediffusionapi.com/docs/a1111schedulers/
|
| 243 |
|
| 244 |
# DPM++ 2M Karras
|
| 245 |
+
# pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
| 246 |
+
# pipe.scheduler.config,
|
| 247 |
+
# use_karras_sigmas=True,
|
| 248 |
+
# )
|
| 249 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
| 250 |
pipe.scheduler.config,
|
| 251 |
+
algorithm_type="sde-dpmsolver++",
|
| 252 |
use_karras_sigmas=True,
|
| 253 |
)
|
| 254 |
|