Fix

model.py

@@ -54,7 +54,8 @@ class Model:
         self.pipe = self.load_pipe(base_model_id, task_name)
 
     def load_pipe(self, base_model_id: str, task_name) -> DiffusionPipeline:
-        if base_model_id == self.base_model_id and task_name == self.task_name:
+        if base_model_id == self.base_model_id and task_name == self.task_name and hasattr(
+                self, 'pipe'):
             return self.pipe
         model_id = CONTROLNET_MODEL_IDS[task_name]
         controlnet = ControlNetModel.from_pretrained(model_id,