Upload download_models.py with huggingface_hub
Browse files- download_models.py +10 -10
download_models.py
CHANGED
@@ -291,26 +291,26 @@ NODE_DISPLAY_NAME_MAPPINGS = {
|
|
291 |
# Define model download configuration
|
292 |
print("Setting up model downloads...")
|
293 |
models_to_download = [
|
294 |
-
# SDXL Checkpoint
|
295 |
{"repo_id": "stabilityai/sdxl-turbo", "filename": "sd_xl_turbo_1.0_fp16.safetensors", "local_path": f"{MODELS_DIR}/checkpoints/dreamshaperXL_turboDpmppSDEKarras.safetensors"},
|
296 |
|
297 |
-
# ControlNet model
|
298 |
{"repo_id": "thibaud/controlnet-openpose-sdxl-1.0", "filename": "pytorch_model.safetensors", "local_path": f"{MODELS_DIR}/controlnet/thibaud_xl_openpose.safetensors"},
|
299 |
|
300 |
-
# IP-Adapter
|
301 |
{"repo_id": "h94/IP-Adapter", "filename": "ip-adapter-plus_sdxl_vit-h.safetensors", "local_path": f"{MODELS_DIR}/ipadapter/ip-adapter_sdxl.safetensors"},
|
302 |
|
303 |
-
# PuLID model files from
|
304 |
-
{"repo_id": "
|
305 |
|
306 |
-
# CLIP Vision
|
307 |
{"repo_id": "h94/IP-Adapter", "filename": "models/image_encoder/model.safetensors", "local_path": f"{MODELS_DIR}/clip_vision/CLIP-ViT-bigG-14-laion2B-39B-b160k.safetensors"},
|
308 |
|
309 |
-
# EVA-CLIP model
|
310 |
-
{"repo_id": "
|
311 |
|
312 |
-
# InsightFace model
|
313 |
-
{"repo_id": "
|
314 |
]
|
315 |
|
316 |
# Try to download each model
|
|
|
291 |
# Define model download configuration
|
292 |
print("Setting up model downloads...")
|
293 |
models_to_download = [
|
294 |
+
# SDXL Checkpoint - Use a public model for now
|
295 |
{"repo_id": "stabilityai/sdxl-turbo", "filename": "sd_xl_turbo_1.0_fp16.safetensors", "local_path": f"{MODELS_DIR}/checkpoints/dreamshaperXL_turboDpmppSDEKarras.safetensors"},
|
296 |
|
297 |
+
# ControlNet model - Use a public model for now
|
298 |
{"repo_id": "thibaud/controlnet-openpose-sdxl-1.0", "filename": "pytorch_model.safetensors", "local_path": f"{MODELS_DIR}/controlnet/thibaud_xl_openpose.safetensors"},
|
299 |
|
300 |
+
# IP-Adapter - Use a public model for now
|
301 |
{"repo_id": "h94/IP-Adapter", "filename": "ip-adapter-plus_sdxl_vit-h.safetensors", "local_path": f"{MODELS_DIR}/ipadapter/ip-adapter_sdxl.safetensors"},
|
302 |
|
303 |
+
# PuLID model files from your backup repository
|
304 |
+
{"repo_id": "Defter77/pulid-models", "filename": "pulid_v1.1.safetensors", "local_path": f"{MODELS_DIR}/pulid/ip-adapter_pulid_sdxl_fp16.safetensors"},
|
305 |
|
306 |
+
# CLIP Vision - Use a public model for now
|
307 |
{"repo_id": "h94/IP-Adapter", "filename": "models/image_encoder/model.safetensors", "local_path": f"{MODELS_DIR}/clip_vision/CLIP-ViT-bigG-14-laion2B-39B-b160k.safetensors"},
|
308 |
|
309 |
+
# EVA-CLIP model from your backup repository
|
310 |
+
{"repo_id": "Defter77/evaclip-models", "filename": "EVA02-CLIP-bigE-14-plus.pt", "local_path": f"{MODELS_DIR}/evaclip/EVA02-CLIP-bigE-14-plus.pt"},
|
311 |
|
312 |
+
# InsightFace model from your backup repository
|
313 |
+
{"repo_id": "Defter77/insightface-models", "filename": "1k3d68.onnx", "local_path": f"{MODELS_DIR}/insightface/1k3d68.onnx"}
|
314 |
]
|
315 |
|
316 |
# Try to download each model
|