Update app.py

app.py

@@ -15,7 +15,7 @@ import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
+model_id = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
 vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 flow_shift = 1.0 #5.0  1.0 for image, 5.0 for 720P, 3.0 for 480P
@@ -50,7 +50,7 @@ CURRENTLY_LOADED_CUSTOM_LORA = None
 # Load the default base LoRA ONCE at startup
 print("Loading base LoRA...")
 CAUSVID_LORA_REPO = "Kijai/WanVideo_comfy"
-CAUSVID_LORA_FILENAME = "Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors"
+CAUSVID_LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
 # try:
 #     causvid_path = hf_hub_download(repo_id=CAUSVID_LORA_REPO, filename=CAUSVID_LORA_FILENAME)
 #     pipe.load_lora_weights(causvid_path, adapter_name=DEFAULT_LORA_NAME)