Spaces:
Running
on
Zero
Running
on
Zero
Rishi Desai
commited on
Commit
·
b5a5ad0
1
Parent(s):
0823773
HF token
Browse files- install.py +5 -1
install.py
CHANGED
@@ -78,6 +78,9 @@ def download_huggingface_models():
|
|
78 |
{"repo_id": "comfyanonymous/flux_text_encoders", "filename": "clip_l.safetensors", "folder": "text_encoders"},
|
79 |
]
|
80 |
|
|
|
|
|
|
|
81 |
# Dictionary mapping repo_ids to specific filenames
|
82 |
filename_mappings = {
|
83 |
"Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro": "Flux_Dev_ControlNet_Union_Pro_ShakkerLabs.safetensors",
|
@@ -89,7 +92,8 @@ def download_huggingface_models():
|
|
89 |
repo_id=model["repo_id"],
|
90 |
filename=model["filename"],
|
91 |
cache_dir=CACHE_PATH,
|
92 |
-
repo_type=model.get("repo_type", "model")
|
|
|
93 |
)
|
94 |
target_dir = os.path.join(MODEL_PATH, model["folder"])
|
95 |
os.makedirs(target_dir, exist_ok=True)
|
|
|
78 |
{"repo_id": "comfyanonymous/flux_text_encoders", "filename": "clip_l.safetensors", "folder": "text_encoders"},
|
79 |
]
|
80 |
|
81 |
+
# Get the Hugging Face token from the environment variable
|
82 |
+
huggingface_token = os.getenv('HUGGINGFACE_TOKEN')
|
83 |
+
|
84 |
# Dictionary mapping repo_ids to specific filenames
|
85 |
filename_mappings = {
|
86 |
"Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro": "Flux_Dev_ControlNet_Union_Pro_ShakkerLabs.safetensors",
|
|
|
92 |
repo_id=model["repo_id"],
|
93 |
filename=model["filename"],
|
94 |
cache_dir=CACHE_PATH,
|
95 |
+
repo_type=model.get("repo_type", "model"),
|
96 |
+
token=huggingface_token
|
97 |
)
|
98 |
target_dir = os.path.join(MODEL_PATH, model["folder"])
|
99 |
os.makedirs(target_dir, exist_ok=True)
|