multimodalart HF Staff commited on
Commit
2630bd2
·
verified ·
1 Parent(s): 05ad421

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -13,13 +13,17 @@ def hf_hub_download_local(repo_id, filename, local_dir, **kwargs):
13
  os.makedirs(local_dir, exist_ok=True)
14
  base_filename = os.path.basename(filename)
15
  target_path = os.path.join(local_dir, base_filename)
16
- shutil.move(downloaded_path, target_path)
 
 
 
 
17
  return target_path
18
 
19
  # --- Model Downloads ---
20
  print("Downloading models from Hugging Face Hub...")
21
- text_encooder_repo = hf_hub_download_local(repo_id="Comfy-Org/Wan_2.1_ComfyUI_repackaged", filename="split_files/text_encoders/umt5_xxl_fp8_e4m3fn_scaled.safetensors", local_dir="models/text_encoders")
22
- print(text_encooder_repo)
23
  hf_hub_download_local(repo_id="Comfy-Org/Wan_2.2_ComfyUI_Repackaged", filename="split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors", local_dir="models/unet")
24
  hf_hub_download_local(repo_id="Comfy-Org/Wan_2.2_ComfyUI_Repackaged", filename="split_files/diffusion_models/wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors", local_dir="models/unet")
25
  hf_hub_download_local(repo_id="Comfy-Org/Wan_2.1_ComfyUI_repackaged", filename="split_files/vae/wan_2.1_vae.safetensors", local_dir="models/vae")
 
13
  os.makedirs(local_dir, exist_ok=True)
14
  base_filename = os.path.basename(filename)
15
  target_path = os.path.join(local_dir, base_filename)
16
+
17
+ if os.path.exists(target_path) or os.path.islink(target_path):
18
+ os.remove(target_path)
19
+
20
+ os.symlink(downloaded_path, target_path)
21
  return target_path
22
 
23
  # --- Model Downloads ---
24
  print("Downloading models from Hugging Face Hub...")
25
+ text_encoder_repo = hf_hub_download_local(repo_id="Comfy-Org/Wan_2.1_ComfyUI_repackaged", filename="split_files/text_encoders/umt5_xxl_fp8_e4m3fn_scaled.safetensors", local_dir="models/text_encoders")
26
+ print(text_encoder_repo)
27
  hf_hub_download_local(repo_id="Comfy-Org/Wan_2.2_ComfyUI_Repackaged", filename="split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors", local_dir="models/unet")
28
  hf_hub_download_local(repo_id="Comfy-Org/Wan_2.2_ComfyUI_Repackaged", filename="split_files/diffusion_models/wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors", local_dir="models/unet")
29
  hf_hub_download_local(repo_id="Comfy-Org/Wan_2.1_ComfyUI_repackaged", filename="split_files/vae/wan_2.1_vae.safetensors", local_dir="models/vae")