Spaces:
Running
Running
fix last
Browse files- Image-Morpher/main.py +1 -1
Image-Morpher/main.py
CHANGED
|
@@ -124,7 +124,7 @@ pipeline.enable_xformers_memory_efficient_attention()
|
|
| 124 |
|
| 125 |
from accelerate import init_empty_weights
|
| 126 |
with init_empty_weights():
|
| 127 |
-
pipeline.to("cuda"
|
| 128 |
|
| 129 |
# Add these AFTER device movement
|
| 130 |
torch.backends.cudnn.benchmark = True # finds efficient convolution algo by running short benchmark, minimal speed-up.
|
|
|
|
| 124 |
|
| 125 |
from accelerate import init_empty_weights
|
| 126 |
with init_empty_weights():
|
| 127 |
+
pipeline.to("cuda")
|
| 128 |
|
| 129 |
# Add these AFTER device movement
|
| 130 |
torch.backends.cudnn.benchmark = True # finds efficient convolution algo by running short benchmark, minimal speed-up.
|