rahul7star commited on
Commit
7d3cca8
·
verified ·
1 Parent(s): e743b21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -11,9 +11,8 @@ from PIL import Image
11
  import random
12
 
13
  MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
14
-
15
- LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
16
- LORA_FILENAME = "FusionX_LoRa/Wan2.1_I2V_14B_FusionX_LoRA.safetensors"
17
 
18
  image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
19
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
 
11
  import random
12
 
13
  MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
14
+ LORA_REPO_ID = "Kijai/WanVideo_comfy"
15
+ LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
 
16
 
17
  image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
18
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)