gokaygokay commited on
Commit
5aa2dd1
1 Parent(s): 01cb633

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -69,8 +69,8 @@ pipe_cyber.unet.set_attn_processor(AttnProcessor2_0())
69
 
70
  # Define samplers
71
  samplers = {
72
- "Euler a": EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config),
73
- "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
74
  }
75
 
76
  DEFAULT_POSITIVE_PREFIX = "score_9, score_8_up, score_7_up, BREAK"
 
69
 
70
  # Define samplers
71
  samplers = {
72
+ "Euler a": EulerAncestralDiscreteScheduler.from_config(pipe_pony.scheduler.config),
73
+ "DPM++ SDE Karras": DPMSolverSDEScheduler.from_config(pipe_pony.scheduler.config, use_karras_sigmas=True)
74
  }
75
 
76
  DEFAULT_POSITIVE_PREFIX = "score_9, score_8_up, score_7_up, BREAK"