Rishi Desai commited on
Commit
b5a5ad0
·
1 Parent(s): 0823773
Files changed (1) hide show
  1. 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)