fix

SUPIR/util.py

@@ -32,14 +32,16 @@ def create_model(config_path):
 
 def resolve_ckpt_path(path_or_hub):
     if os.path.exists(path_or_hub):
-        return path_or_hub  # local path
-    if "/" in path_or_hub and path_or_hub.endswith(".ckpt"):
-        # Assume format: repo_id/path/to/file.ckpt
+        return path_or_hub  # Local path
+
+    # Handle Hugging Face Hub repo/file pattern
+    if "/" in path_or_hub and path_or_hub.endswith((".ckpt", ".safetensors", ".bin")):
         parts = path_or_hub.split("/")
         repo_id = "/".join(parts[:2])
         filename = "/".join(parts[2:])
         return hf_hub_download(repo_id=repo_id, filename=filename)
-    return path_or_hub  # fallback
+
+    raise FileNotFoundError(f"Could not resolve checkpoint path: {path_or_hub}")
 
 def create_SUPIR_model(config_path, SUPIR_sign=None, load_default_setting=False):
     config = OmegaConf.load(config_path)