Defter77 commited on
Commit
e460c61
·
verified ·
1 Parent(s): b329109

Upload download_models.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 f5aiteam/pullid
304
- {"repo_id": "f5aiteam/pullid", "filename": "pulid_v1.1.safetensors", "local_path": f"{MODELS_DIR}/pulid/ip-adapter_pulid_sdxl_fp16.safetensors"},
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": "QuanSun/EVA-CLIP", "filename": "EVA02-CLIP-bigE-14-plus.pt", "local_path": f"{MODELS_DIR}/evaclip/EVA02-CLIP-bigE-14-plus.pt"},
311
 
312
- # InsightFace model (buffalo_l model for facial landmark detection)
313
- {"repo_id": "hanamizuki-ai/insightface-releases", "filename": "buffalo_l/1k3d68.onnx", "local_path": f"{MODELS_DIR}/insightface/1k3d68.onnx"}
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