ford442 commited on
Commit
2b95ba8
·
1 Parent(s): cb31091

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -130,7 +130,7 @@ def load_and_prepare_model(model_id):
130
  sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear",use_karras_sigmas=True, algorithm_type="dpmsolver++")
131
  #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
132
  #pipe.to('cuda')
133
- pipe.scheduler = sched
134
  #pipe.vae=vae
135
  #pipe.vae=pipeX.vae
136
  # pipe.scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
130
  sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear",use_karras_sigmas=True, algorithm_type="dpmsolver++")
131
  #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
132
  #pipe.to('cuda')
133
+ #pipe.scheduler = sched
134
  #pipe.vae=vae
135
  #pipe.vae=pipeX.vae
136
  # pipe.scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)