3v324v23 commited on
Commit
a4a38ad
·
1 Parent(s): 5d0466b

change realistic to DPM++ 2M SDE Karras

Browse files
Files changed (1) hide show
  1. 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