fixes pipe to cpu

Capsize-Games · Jan 9, 2024 · dd9bb6a · dd9bb6a
1 parent a9d4a21
commit dd9bb6a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/airunner/aihandler/mixins/memory_efficient_mixin.py b/src/airunner/aihandler/mixins/memory_efficient_mixin.py
@@ -232,15 +232,17 @@ def move_pipe_to_cpu(self):
         if not self.pipe:
             return
         try:
-            self.pipe.to("cpu", self.data_type).float32()
+            self.pipe.to("cpu", self.data_type)
         except NotImplementedError:
             logger.warning("Not implemented error when moving to cpu")
 
         if hasattr(self.pipe, "controlnet"):
             try:
-                self.pipe.controlnet.to("cpu", self.data_type).float32()
+                self.pipe.controlnet.to("cpu", self.data_type)
             except NotImplementedError:
                 logger.warning("Not implemented error when moving to cpu")
+            except AttributeError:
+                pass
 
     def apply_cpu_offload(self):
         if self.cpu_offload_applied == self.enable_model_cpu_offload: