Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -157,9 +157,9 @@ def load_and_prepare_model(model_id):
|
|
157 |
pipe.to(device)
|
158 |
pipe.to(torch.bfloat16)
|
159 |
|
160 |
-
apply_hidiffusion(pipe)
|
161 |
|
162 |
-
pipe.unet.set_default_attn_processor()
|
163 |
pipe.vae.set_default_attn_processor()
|
164 |
|
165 |
print(f'Pipeline: ')
|
|
|
157 |
pipe.to(device)
|
158 |
pipe.to(torch.bfloat16)
|
159 |
|
160 |
+
#apply_hidiffusion(pipe)
|
161 |
|
162 |
+
#pipe.unet.set_default_attn_processor()
|
163 |
pipe.vae.set_default_attn_processor()
|
164 |
|
165 |
print(f'Pipeline: ')
|