fix: tensors

app.py

@@ -5,21 +5,23 @@ from huggingface_hub import hf_hub_download
 
 def resolve_hf_path(path):
     if isinstance(path, str) and path.startswith("hf://"):
-        # The expected format is: hf://{repo_id}@{filename}
         parts = path[len("hf://"):].split("@")
-        if len(parts) != 2:
+        if len(parts) == 1:
+            repo_id = parts[0]
+            filename = None
+        elif len(parts) == 2:
+            repo_id, filename = parts
+        else:
             raise ValueError(f"Invalid HF URI format: {path}")
         token = os.environ.get("HUGGINGFACE_HUB_TOKEN")
-        repo_id = parts[0]
-        filename = parts[1]
-        print(f"Downloading {filename} from {repo_id} ...")
         if token is None:
             raise ValueError("HUGGINGFACE_HUB_TOKEN environment variable not set!")
-        local_path = hf_hub_download(repo_id=repo_id, filename=filename, token=token)
+        # If filename is provided, download that file; otherwise, download the whole repo snapshot.
+        local_path = hf_hub_download(repo_id=repo_id, filename=filename, token=token) if filename else hf_hub_download(repo_id=repo_id, token=token)
         return local_path
     return path
 
-os.environ["FLUX_FILL_PATH"] = "hf://black-forest-labs/FLUX.1-Fill-dev@ae.safetensors"
+os.environ["FLUX_FILL_PATH"] = "hf://black-forest-labs/FLUX.1-Fill-dev"
 os.environ["PORTRAIT_MODEL_PATH"] = "ms://iic/ACE_Plus@portrait/comfyui_portrait_lora64.safetensors"
 os.environ["SUBJECT_MODEL_PATH"] = "ms://iic/ACE_Plus@subject/comfyui_subject_lora16.safetensors"
 os.environ["LOCAL_MODEL_PATH"] = "ms://iic/ACE_Plus@local_editing/comfyui_local_lora16.safetensors"