ford442 commited on
Commit
9de4246
·
verified ·
1 Parent(s): 343d6c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -148,7 +148,7 @@ def load_and_prepare_model(model_id):
148
  #pipe.scheduler = DPMSolverMultistepScheduler.from_pretrained('SG161222/RealVisXL_V5.0', subfolder='scheduler', algorithm_type='sde-dpmsolver++')
149
  #pipe.vae = vaeX
150
  #pipe.unet = unetX
151
- #pipe.vae.do_resize=False
152
  #pipe.scheduler = sched
153
  #pipe.vae=vae.to(torch.bfloat16)
154
  #pipe.unet=pipeX.unet
@@ -172,13 +172,6 @@ def load_and_prepare_model(model_id):
172
  #print(f'UNET: {pipe.unet}')
173
  pipe.watermark=None
174
  pipe.safety_checker=None
175
- #pipe.to(torch.device("cuda:0"))
176
- #pipe.vae.to(torch.bfloat16)
177
- #pipe.to(device, torch.bfloat16)
178
- #del pipeX
179
- #sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", algorithm_type="dpmsolver++")
180
- #sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="linear", algorithm_type="dpmsolver++")
181
- #sched = DDIMScheduler.from_config(pipe.scheduler.config)
182
  return pipe
183
 
184
  # Preload and compile both models
 
148
  #pipe.scheduler = DPMSolverMultistepScheduler.from_pretrained('SG161222/RealVisXL_V5.0', subfolder='scheduler', algorithm_type='sde-dpmsolver++')
149
  #pipe.vae = vaeX
150
  #pipe.unet = unetX
151
+ pipe.vae.do_resize=False
152
  #pipe.scheduler = sched
153
  #pipe.vae=vae.to(torch.bfloat16)
154
  #pipe.unet=pipeX.unet
 
172
  #print(f'UNET: {pipe.unet}')
173
  pipe.watermark=None
174
  pipe.safety_checker=None
 
 
 
 
 
 
 
175
  return pipe
176
 
177
  # Preload and compile both models