wan2-1-fast-savingvram

Runtime error

adptbyt commited on Jun 8

Commit

70b7643

verified ·

1 Parent(s): 3f6fe64

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import random
 MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
 LORA_REPO_ID = "Kijai/WanVideo_comfy"
-LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32_v2.safetensors"
 image_encoder = CLIPVisionModel.from_pretrained(
     MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float16

 MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
 LORA_REPO_ID = "Kijai/WanVideo_comfy"
+LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32_v1_5_no_first_block.safetensors"
 image_encoder = CLIPVisionModel.from_pretrained(
     MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float16