Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -141,6 +141,8 @@ def load_and_prepare_model():
|
|
141 |
#pipe.vae=vaeX
|
142 |
pipe.to(device=device, dtype=torch.bfloat16)
|
143 |
pipe.vae=vaeX
|
|
|
|
|
144 |
#pipe.vae.to(device=device, dtype=torch.bfloat16)
|
145 |
#pipe.vae.do_resize=False
|
146 |
#pipe.vae.do_rescale=False
|
|
|
141 |
#pipe.vae=vaeX
|
142 |
pipe.to(device=device, dtype=torch.bfloat16)
|
143 |
pipe.vae=vaeX
|
144 |
+
pipe.vae.enable_tiling()
|
145 |
+
pipe.vae.fuse_qkv_projections()
|
146 |
#pipe.vae.to(device=device, dtype=torch.bfloat16)
|
147 |
#pipe.vae.do_resize=False
|
148 |
#pipe.vae.do_rescale=False
|