ford442 commited on
Commit
5b58b56
·
verified ·
1 Parent(s): 4536b36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -153,9 +153,9 @@ def load_and_prepare_model(model_id):
153
  #pipe.unet = unetX
154
  pipe.scheduler = sched
155
  pipe.vae.do_resize=False
156
- pipe.vae.do_rescale=False
157
  pipe.vae.do_convert_rgb=True
158
- pipe.vae.vae_scale_factor=1
159
  #pipe.scheduler = sched
160
  #pipe.vae=vae.to(torch.bfloat16)
161
  #pipe.unet=pipeX.unet
@@ -228,7 +228,7 @@ def uploadNote():
228
  f.write(f"Model UNET: default to bfloat before cuda then attn_proc \n")
229
  f.write(f"Model HiDiffusion OFF \n")
230
  f.write(f"Model do_resize OFF \n")
231
- f.write(f"Model vae_scale_factor 1 / do-rescale=False \n")
232
  upload_to_ftp(filename)
233
 
234
  @spaces.GPU(duration=30)
 
153
  #pipe.unet = unetX
154
  pipe.scheduler = sched
155
  pipe.vae.do_resize=False
156
+ #pipe.vae.do_rescale=False
157
  pipe.vae.do_convert_rgb=True
158
+ pipe.vae.vae_scale_factor=4
159
  #pipe.scheduler = sched
160
  #pipe.vae=vae.to(torch.bfloat16)
161
  #pipe.unet=pipeX.unet
 
228
  f.write(f"Model UNET: default to bfloat before cuda then attn_proc \n")
229
  f.write(f"Model HiDiffusion OFF \n")
230
  f.write(f"Model do_resize OFF \n")
231
+ f.write(f"Model vae_scale_factor 4 \n")
232
  upload_to_ftp(filename)
233
 
234
  @spaces.GPU(duration=30)