ControlNet-with-Anything-v4

Running

hysts HF staff commited on Mar 10, 2023

Commit

cbc964f

•

1 Parent(s): fef86c6

Update

Files changed (1) hide show

model.py CHANGED Viewed

@@ -90,7 +90,8 @@ class Model:
     def load_controlnet_weight(self, task_name: str) -> None:
         if task_name == self.task_name:
             return
-        del self.pipe.controlnet
         torch.cuda.empty_cache()
         gc.collect()
         model_id = CONTROLNET_MODEL_IDS[task_name]

     def load_controlnet_weight(self, task_name: str) -> None:
         if task_name == self.task_name:
             return
+        if hasattr(self.pipe, 'controlnet'):
+            del self.pipe.controlnet
         torch.cuda.empty_cache()
         gc.collect()
         model_id = CONTROLNET_MODEL_IDS[task_name]