pr 1: fix finding the VAE

comfyui/ComfyUI-SeedVR2-1.4B-TinyVAE/__init__.py

@@ -51,6 +51,7 @@ Environment
 
 import logging
 import os
+import sys
 
 log = logging.getLogger(__name__)
 
@@ -81,24 +82,38 @@ COMFY_PRE_SCALES = True        # ComfyUI already applied it before decode_ (veri
 
 _WEIGHT_NAME = "ema_vae_distill_fp16.safetensors"
 
+try:
+    import folder_paths
+except ImportError:
+    comfy_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+    sys.path.insert(0, comfy_path)
+    try:
+        import folder_paths
+    except ImportError:
+        folder_paths = None
+        log.warning("Could not import folder_paths, falling back to local path resolution")
 
 def _find_weights():
     if os.environ.get("SEEDVR2_TINYVAE_PATH"):
         return os.environ["SEEDVR2_TINYVAE_PATH"]
     here = os.path.dirname(os.path.abspath(__file__))
+    
     cands = [
         os.path.join(here, _WEIGHT_NAME),
-        os.path.join(here, "..", _WEIGHT_NAME),
-        os.path.join(here, "..", "..", _WEIGHT_NAME),
+        os.path.join(os.path.dirname(here), _WEIGHT_NAME),
+        os.path.join(folder_paths.get_folder_paths("vae")[0], _WEIGHT_NAME) if folder_paths and folder_paths.get_folder_paths("vae") else None,
+        *([os.path.join(p, _WEIGHT_NAME) for p in folder_paths.get_folder_paths("vae")] if folder_paths else []),
+        os.path.join(here, "models", _WEIGHT_NAME),
     ]
-    try:                                                  # ComfyUI's own vae folder
-        import folder_paths
-        cands += [os.path.join(p, _WEIGHT_NAME) for p in folder_paths.get_folder_paths("vae")]
-    except Exception:
-        pass
+    
+    cands = [c for c in cands if c]
+    
     for c in cands:
         if os.path.exists(c):
+            log.info(f"SeedVR2 TinyVAE: Found weights at {c}")
             return os.path.abspath(c)
+    
+    log.warning(f"SeedVR2 TinyVAE: Could not find {_WEIGHT_NAME} in any of the searched locations: {cands}")
     return None