Update src/flux/xflux_pipeline.py

src/flux/xflux_pipeline.py

@@ -55,7 +55,7 @@ class XFluxPipeline:
         self.controlnet_loaded = False
         self.ip_loaded = False
 
-    def to(*args, **kwargs):
+    def to(self, *args, **kwargs):
         self.model.to(*args, **kwargs)
         self.clip.to(*args, **kwargs)
         self.t5.to(*args, **kwargs)