rahul7star commited on
Commit
712c3a2
·
verified ·
1 Parent(s): f0449c9

Update app_t2v.py

Browse files
Files changed (1) hide show
  1. app_t2v.py +3 -1
app_t2v.py CHANGED
@@ -5,6 +5,7 @@ sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
5
  import gradio as gr
6
  import torch
7
  from huggingface_hub import snapshot_download
 
8
  from PIL import Image
9
  import random
10
  import numpy as np
@@ -90,7 +91,8 @@ wan_pipeline = wan.WanTI2V(
90
 
91
 
92
  LORA_REPO_ID = "Kijai/WanVideo_comfy"
93
- LORA_FILENAME = "Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensorscausvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
 
94
  wan_pipeline.load_lora_weights(causvid_path, adapter_name="causvid_lora")
95
  wan_pipeline.set_adapters(["causvid_lora"], adapter_weights=[0.95])
96
  wan_pipeline.fuse_lora()"
 
5
  import gradio as gr
6
  import torch
7
  from huggingface_hub import snapshot_download
8
+ from huggingface_hub import hf_hub_download
9
  from PIL import Image
10
  import random
11
  import numpy as np
 
91
 
92
 
93
  LORA_REPO_ID = "Kijai/WanVideo_comfy"
94
+ LORA_FILENAME = "Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors"
95
+ causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
96
  wan_pipeline.load_lora_weights(causvid_path, adapter_name="causvid_lora")
97
  wan_pipeline.set_adapters(["causvid_lora"], adapter_weights=[0.95])
98
  wan_pipeline.fuse_lora()"