Hugging Face's logo

Spaces:
rahul7star
/
wan-fusionx-lora
Running on Zero

App Files Community
rahul7star commited on
Commit
fa56b78
·
verified ·
1 Parent(s): d52fab5

Update app_lora.py

Browse files
Files changed (1) hide show
  1. app_lora.py +2 -1
app_lora.py CHANGED
@@ -14,7 +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
 
19
  image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
20
  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)