Spaces:
Running
on
A100
Running
on
A100
fix typo
Browse files
pipelines/controlnetLoraSD15.py
CHANGED
@@ -172,7 +172,7 @@ class Pipeline:
|
|
172 |
pipe.set_progress_bar_config(disable=True)
|
173 |
pipe.to(device=device, dtype=torch_dtype).to(device)
|
174 |
if device.type != "mps":
|
175 |
-
|
176 |
|
177 |
if psutil.virtual_memory().total < 64 * 1024**3:
|
178 |
pipe.enable_attention_slicing()
|
|
|
172 |
pipe.set_progress_bar_config(disable=True)
|
173 |
pipe.to(device=device, dtype=torch_dtype).to(device)
|
174 |
if device.type != "mps":
|
175 |
+
pipe.unet.to(memory_format=torch.channels_last)
|
176 |
|
177 |
if psutil.virtual_memory().total < 64 * 1024**3:
|
178 |
pipe.enable_attention_slicing()
|