Spaces:
Running
on
Zero
Running
on
Zero
Update optimized.py
Browse files- optimized.py +1 -1
optimized.py
CHANGED
|
@@ -34,7 +34,7 @@ def self_attention_slicing(module, slice_size=3):
|
|
| 34 |
|
| 35 |
return output
|
| 36 |
return sliced_attention
|
| 37 |
-
|
| 38 |
|
| 39 |
huggingface_token = os.getenv("HUGGINFACE_TOKEN")
|
| 40 |
good_vae = AutoencoderKL.from_pretrained(
|
|
|
|
| 34 |
|
| 35 |
return output
|
| 36 |
return sliced_attention
|
| 37 |
+
device = "cuda"
|
| 38 |
|
| 39 |
huggingface_token = os.getenv("HUGGINFACE_TOKEN")
|
| 40 |
good_vae = AutoencoderKL.from_pretrained(
|