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.
|