rahul7star commited on
Commit
b29fa42
·
verified ·
1 Parent(s): fa56b78

Update app_lora.py

Browse files
Files changed (1) hide show
  1. app_lora.py +2 -2
app_lora.py CHANGED
@@ -14,8 +14,8 @@ import random
14
  MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
15
 
16
  LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
17
- #LORA_FILENAME = "FusionX_LoRa/Wan2.1_I2V_14B_FusionX_LoRA.safetensors"
18
- LORA_FILENAME = "Wan14Bi2vFusioniX_fp16.safetensors"
19
 
20
  image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
21
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
 
14
  MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
15
 
16
  LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
17
+ LORA_FILENAME = "FusionX_LoRa/Wan2.1_I2V_14B_FusionX_LoRA.safetensors"
18
+ #LORA_FILENAME = "Wan14Bi2vFusioniX_fp16.safetensors"
19
 
20
  image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
21
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)