Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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
|
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)
|