Spaces:
Paused
Paused
Update app_t2v.py
Browse files- 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.
|
|
|
|
| 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()"
|