Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
17 |
return target_path
|
18 |
|
19 |
# --- Model Downloads ---
|
20 |
print("Downloading models from Hugging Face Hub...")
|
21 |
-
|
22 |
-
print(
|
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")
|