ford442 commited on
Commit
e3e00ed
·
1 Parent(s): e442a24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -121,7 +121,7 @@ def load_and_prepare_model(model_id):
121
  )
122
  #sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2",use_karras_sigmas=True, algorithm_type="dpmsolver++")
123
  sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2", algorithm_type="dpmsolver++")
124
-
125
  pipe.scheduler=sched
126
  #pipe.to(device=device, dtype=torch.bfloat16)
127
  pipe.to(device=device)
 
121
  )
122
  #sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2",use_karras_sigmas=True, algorithm_type="dpmsolver++")
123
  sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2", algorithm_type="dpmsolver++")
124
+ #sched = DDIMScheduler.from_config(pipe.scheduler.config,)
125
  pipe.scheduler=sched
126
  #pipe.to(device=device, dtype=torch.bfloat16)
127
  pipe.to(device=device)