Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -161,7 +161,10 @@ def load_and_prepare_model(model_id):
|
|
| 161 |
|
| 162 |
pipe.to(device)
|
| 163 |
pipe.to(torch.bfloat16)
|
| 164 |
-
|
|
|
|
|
|
|
|
|
|
| 165 |
#pipe.to(torch.bfloat16)
|
| 166 |
print(f'Pipeline: ')
|
| 167 |
print(f'_optional_components: {pipe._optional_components}')
|
|
|
|
| 161 |
|
| 162 |
pipe.to(device)
|
| 163 |
pipe.to(torch.bfloat16)
|
| 164 |
+
|
| 165 |
+
pipe.unet.set_default_attn_processor()
|
| 166 |
+
pipe.vae.set_default_attn_processor()
|
| 167 |
+
|
| 168 |
#pipe.to(torch.bfloat16)
|
| 169 |
print(f'Pipeline: ')
|
| 170 |
print(f'_optional_components: {pipe._optional_components}')
|