dont unload controlnets when changhing ip adapter

painebenjamin · Jan 19, 2024 · c3ded9e · c3ded9e
1 parent 1a9da83
commit c3ded9e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/python/enfugue/diffusion/manager.py b/src/python/enfugue/diffusion/manager.py
@@ -5359,12 +5359,13 @@ def memoize_callback(images: List[PIL.Image.Image]) -> None:
                     if hasattr(self, "_pipeline"):
                         if self._ip_adapter_model is None:
                             if self._pipeline.ip_adapter_loaded:
-                                self.unload_pipeline("disabling IP adapter")
+                                self.unload_pipeline("disabling IP adapter", False)
                         else:
                             if not self._pipeline.ip_adapter_loaded:
-                                self.unload_pipeline("enabling IP adapter")
+                                self.unload_pipeline("enabling IP adapter", False)
                             elif self.ip_adapter.model != self._ip_adapter_model:
-                                self.unload_pipeline("changing IP adapter model")
+                                self.unload_pipeline("changing IP adapter model", False)
+
                     pipe = self.pipeline # type: ignore
 
                 # Check refining settings