Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -121,7 +121,7 @@ def load_and_prepare_model():
|
|
121 |
text_encoder=None,
|
122 |
text_encoder_2=None,
|
123 |
vae=None,
|
124 |
-
|
125 |
)
|
126 |
|
127 |
'''
|
@@ -141,8 +141,8 @@ def load_and_prepare_model():
|
|
141 |
#pipe.vae=vaeX
|
142 |
pipe.to(device=device, dtype=torch.bfloat16)
|
143 |
pipe.vae=vaeX
|
144 |
-
|
145 |
-
|
146 |
#pipe.vae.to(device=device, dtype=torch.bfloat16)
|
147 |
#pipe.vae.do_resize=False
|
148 |
#pipe.vae.do_rescale=False
|
|
|
121 |
text_encoder=None,
|
122 |
text_encoder_2=None,
|
123 |
vae=None,
|
124 |
+
# unet=None,
|
125 |
)
|
126 |
|
127 |
'''
|
|
|
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
|