Update gradio_app/all_models.py

gradio_app/all_models.py

@@ -1,22 +1,22 @@
-import torch
-from scripts.sd_model_zoo import load_common_sd15_pipe
-from diffusers import StableDiffusionControlNetImg2ImgPipeline, StableDiffusionPipeline
-
-
-class MyModelZoo:
-    _pipe_disney_controlnet_lineart_ipadapter_i2i: StableDiffusionControlNetImg2ImgPipeline = None
-    
-    base_model = "runwayml/stable-diffusion-v1-5"
-
-    def __init__(self, base_model=None) -> None:
-        if base_model is not None:
-            self.base_model = base_model
-
-    @property
-    def pipe_disney_controlnet_tile_ipadapter_i2i(self):
-        return self._pipe_disney_controlnet_lineart_ipadapter_i2i
-    
-    def init_models(self):
-        self._pipe_disney_controlnet_lineart_ipadapter_i2i = load_common_sd15_pipe(base_model=self.base_model, ip_adapter=True, plus_model=False, controlnet="./ckpt/controlnet-tile", pipeline_class=StableDiffusionControlNetImg2ImgPipeline)
-
-model_zoo = MyModelZoo()
+import torch
+from scripts.sd_model_zoo import load_common_sd15_pipe
+from diffusers import StableDiffusionControlNetImg2ImgPipeline, StableDiffusionPipeline
+
+
+class MyModelZoo:
+    _pipe_disney_controlnet_lineart_ipadapter_i2i: StableDiffusionControlNetImg2ImgPipeline = None
+    
+    base_model = "benjamin-paine/stable-diffusion-v1-5"
+
+    def __init__(self, base_model=None) -> None:
+        if base_model is not None:
+            self.base_model = base_model
+
+    @property
+    def pipe_disney_controlnet_tile_ipadapter_i2i(self):
+        return self._pipe_disney_controlnet_lineart_ipadapter_i2i
+    
+    def init_models(self):
+        self._pipe_disney_controlnet_lineart_ipadapter_i2i = load_common_sd15_pipe(base_model=self.base_model, ip_adapter=True, plus_model=False, controlnet="./ckpt/controlnet-tile", pipeline_class=StableDiffusionControlNetImg2ImgPipeline)
+
+model_zoo = MyModelZoo()