nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
SolitaryThinker
2025-08-04 05:30:09 +00:00
parent 242e38e98d
commit 8b2f2b556c
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/inference/gradio/gradio_frontend.py
+6 -2
View File
@@ -20,7 +20,10 @@ from fastvideo.configs.sample.base import SamplingParam
MODEL_PATH_MAPPING = {
"FastWan2.1-T2V-1.3B": "FastVideo/FastWan2.1-T2V-1.3B-Diffusers",
"FastWan2.1-T2V-14B": "FastVideo/FastWan2.1-T2V-14B-Diffusers",
"FastWan2.2-TI2V-5B": "FastVideo/FastWan2.2-TI2V-5B-Diffusers"
"FastWan2.2-TI2V-5B": "FastVideo/FastWan2.2-TI2V-5B-Diffusers",
"Wan2.1-T2V-1.3B": "Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
"Wan2.1-T2V-14B": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
"Wan2.2-TI2V-5B": "Wan-AI/Wan2.2-TI2V-5B-Diffusers"
}
@@ -441,8 +444,9 @@ def create_gradio_interface(backend_url: str, default_params: dict[str, Sampling
model_selection = gr.Dropdown(
choices=[
"FastWan2.1-T2V-1.3B",
"FastWan2.1-T2V-14B",
"FastWan2.2-TI2V-5B",
"Wan2.1-T2V-1.3B",
"Wan2.2-TI2V-5B",
# "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers (Image-to-Video)" # I2V functionality commented out
],
value="FastWan2.1-T2V-1.3B",
examples/inference/gradio/ray_serve_backend.py
+1 -1
View File
@@ -106,7 +106,7 @@ class T2VModelDeployment:
os.makedirs(self.output_path, exist_ok=True)
# Delay helps avoid GPU contention when many replicas start at once
time.sleep(5)
# time.sleep(5)
# Ensure correct attention backend for FastVideo
if "FastVideo" in self.model_path: