JiaxinGe
/

Diffusers-BAGEL

Model card Files Files and versions

para-lost commited on Jul 25, 2025

Commit

3ad4866

·

1 Parent(s): b964e64

pipeline update

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -7041,7 +7041,7 @@ class BagelPipeline(DiffusionPipeline):
         return out
     def to(self, device):
-        super().to(device)          # moves registered modules
         if hasattr(self, "_inferencer"):
             self._inferencer.to(device)
         return self

         return out
     def to(self, device):
+        super().to(device)
         if hasattr(self, "_inferencer"):
             self._inferencer.to(device)
         return self