Spaces:
Paused
Paused
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)
|