Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -91,6 +91,8 @@ text_encoder.requires_grad_(False)
|
|
| 91 |
unet.requires_grad_(False)
|
| 92 |
controlnet.requires_grad_(False)
|
| 93 |
|
|
|
|
|
|
|
| 94 |
unet.enable_xformers_memory_efficient_attention()
|
| 95 |
controlnet.enable_xformers_memory_efficient_attention()
|
| 96 |
|
|
|
|
| 91 |
unet.requires_grad_(False)
|
| 92 |
controlnet.requires_grad_(False)
|
| 93 |
|
| 94 |
+
unet.to("cuda")
|
| 95 |
+
controlnet.to("cuda")
|
| 96 |
unet.enable_xformers_memory_efficient_attention()
|
| 97 |
controlnet.enable_xformers_memory_efficient_attention()
|
| 98 |
|