superdiff
/

superdiff-sdxl-v1-0

SuperDiffSDXLPipeline

Model card Files Files and versions Community

mskrt commited on Mar 6

Commit

2b7c5c6

·

verified ·

1 Parent(s): dd1fefc

Upload pipeline.py

Files changed (1) hide show

pipeline.py +3 -3

pipeline.py CHANGED Viewed

@@ -102,9 +102,9 @@ class SuperDiffSDXLPipeline(DiffusionPipeline, ConfigMixin):
         dtype = torch.float16
         vae.to(device)
-        unet.to(device)
-        text_encoder.to(device)
-        text_encoder_2.to(device)
         self.register_modules(
             unet=unet,

         dtype = torch.float16
         vae.to(device)
+        unet.to(device, dtype=dtype)
+        text_encoder.to(device, dtype=dtype)
+        text_encoder_2.to(device, dtype=dtype)
         self.register_modules(
             unet=unet,