Update sonic.py

sonic.py

@@ -146,9 +146,24 @@ class Sonic:
         self._load_models(cfg)
         print("Sonic init done")
 
+    def _locate_diffusers_dir(root: str) -> str:
+        """
+        root 아래에서 model_index.json 또는 config.json 이 존재하는
+        디렉터리를 찾아 반환. (snapshots/<sha>/ … 형식 대응)
+        """
+        for cur, _dirs, files in os.walk(root):
+            if {"model_index.json", "config.json"} & set(files):
+                return cur
+        raise FileNotFoundError(
+            f"[ERROR] diffusers model files(model_index.json/config.json) "
+            f"not found under {root}"
+        )
+
+    
     # --------------------------------------------- load all networks
     def _load_models(self, cfg):
         dtype = {"fp16": torch.float16, "fp32": torch.float32, "bf16": torch.bfloat16}[cfg.weight_dtype]
+        diff_root = _locate_diffusers_dir(self.diffusers_root)   # ★★ 핵심 추가
 
         vae   = AutoencoderKLTemporalDecoder.from_pretrained(self.diffusers_root, subfolder="vae", variant="fp16")
         sched = EulerDiscreteScheduler.from_pretrained(self.diffusers_root, subfolder="scheduler")