StreamMultiDiffusion

Runtime error

ironjr commited on Apr 20

Commit

1856601

•

1 Parent(s): e533760

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -149,6 +149,7 @@ class StreamMultiDiffusion(nn.Module):
             lora_scale=1.0,
             safe_fusing=False,
         )
         if use_xformers:
             self.pipe.enable_xformers_memory_efficient_attention()

             lora_scale=1.0,
             safe_fusing=False,
         )
+        self.pipe.to(device=self.device, dtype=self.dtype)
         if use_xformers:
             self.pipe.enable_xformers_memory_efficient_attention()