Real-Time-Latent-Consistency-Model-Text-To-Image

Running on A10G

radames commited on Nov 20, 2023

Commit

7b4b3a3

•

1 Parent(s): cf3ff1a

fix typo

Files changed (1) hide show

pipelines/controlnetLoraSD15.py CHANGED Viewed

@@ -172,7 +172,7 @@ class Pipeline:
             pipe.set_progress_bar_config(disable=True)
             pipe.to(device=device, dtype=torch_dtype).to(device)
             if device.type != "mps":
-                self.pipe.unet.to(memory_format=torch.channels_last)
             if psutil.virtual_memory().total < 64 * 1024**3:
                 pipe.enable_attention_slicing()

             pipe.set_progress_bar_config(disable=True)
             pipe.to(device=device, dtype=torch_dtype).to(device)
             if device.type != "mps":
+                pipe.unet.to(memory_format=torch.channels_last)
             if psutil.virtual_memory().total < 64 * 1024**3:
                 pipe.enable_attention_slicing()