Change model repo id

custum_3d_diffusion/custum_modules/unifield_processor.py

@@ -70,7 +70,7 @@ class AttnConfig:
     use_mv_joint_attn: bool = False
     
     # for unet
-    init_unet_path: str = "runwayml/stable-diffusion-v1-5"
+    init_unet_path: str = "stable-diffusion-v1-5/stable-diffusion-v1-5"
     init_num_cls_label: int = 0                         # for initialize
     cls_labels: List[int] = field(default_factory=lambda: [])
     cls_label_type: str = "embedding"

gradio_app/all_models.py

@@ -6,7 +6,7 @@ from diffusers import StableDiffusionControlNetImg2ImgPipeline, StableDiffusionP
 class MyModelZoo:
     _pipe_disney_controlnet_lineart_ipadapter_i2i: StableDiffusionControlNetImg2ImgPipeline = None
     
-    base_model = "benjamin-paine/stable-diffusion-v1-5"
+    base_model = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 
     def __init__(self, base_model=None) -> None:
         if base_model is not None:

scripts/sd_model_zoo.py

@@ -4,7 +4,7 @@ import torch
 from copy import deepcopy
 
 ENABLE_CPU_CACHE = False
-DEFAULT_BASE_MODEL = "benjamin-paine/stable-diffusion-v1-5"
+DEFAULT_BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 
 cached_models = {}  # cache for models to avoid repeated loading, key is model name
 def cache_model(func):