radames commited on
Commit
7b4b3a3
1 Parent(s): cf3ff1a
Files changed (1) hide show
  1. pipelines/controlnetLoraSD15.py +1 -1
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
- self.pipe.unet.to(memory_format=torch.channels_last)
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()