From f61c0685eb604ba02a0ca0981d37bcd80545a8ca Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Mon, 28 Jul 2025 21:13:54 +0000 Subject: [PATCH] lint --- fastvideo/configs/sample/registry.py | 35 +++++++++++++++--------- fastvideo/entrypoints/video_generator.py | 2 +- 2 files changed, 23 insertions(+), 14 deletions(-) diff --git a/fastvideo/configs/sample/registry.py b/fastvideo/configs/sample/registry.py index e750b4db4..c72a12854 100644 --- a/fastvideo/configs/sample/registry.py +++ b/fastvideo/configs/sample/registry.py @@ -6,12 +6,11 @@ from typing import Any from fastvideo.configs.sample.hunyuan import (FastHunyuanSamplingParam, HunyuanSamplingParam) from fastvideo.configs.sample.stepvideo import StepVideoT2VSamplingParam -from fastvideo.configs.sample.wan import (FastWanT2V480PConfig, +from fastvideo.configs.sample.wan import (Wan2_1_Fun_1_3B_InP_SamplingParam, WanI2V_14B_480P_SamplingParam, WanI2V_14B_720P_SamplingParam, WanT2V_1_3B_SamplingParam, - WanT2V_14B_SamplingParam, - Wan2_1_Fun_1_3B_InP_SamplingParam) + WanT2V_14B_SamplingParam) from fastvideo.logger import init_logger from fastvideo.utils import (maybe_download_model_index, verify_model_config_and_directory) @@ -19,16 +18,26 @@ from fastvideo.utils import (maybe_download_model_index, logger = init_logger(__name__) # Registry maps specific model weights to their config classes SAMPLING_PARAM_REGISTRY: dict[str, Any] = { - "FastVideo/FastHunyuan-diffusers": FastHunyuanSamplingParam, - "hunyuanvideo-community/HunyuanVideo": HunyuanSamplingParam, - "Wan-AI/Wan2.1-T2V-1.3B-Diffusers": WanT2V_1_3B_SamplingParam, - "Wan-AI/Wan2.1-T2V-14B-Diffusers": WanT2V_14B_SamplingParam, - "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers": WanI2V_14B_480P_SamplingParam, - "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers": WanI2V_14B_720P_SamplingParam, - "weizhou03/Wan2.1-Fun-1.3B-InP-Diffusers": Wan2_1_Fun_1_3B_InP_SamplingParam, - "FastVideo/stepvideo-t2v-diffusers": StepVideoT2VSamplingParam, - "FastVideo/FastWan2.1-T2V-1.3B-Diffusers": Wan2_1_Fun_1_3B_InP_SamplingParam, - "FastVideo/FastWan2.1-T2V-14B-480P-Diffusers": Wan2_1_Fun_1_3B_InP_SamplingParam, + "FastVideo/FastHunyuan-diffusers": + FastHunyuanSamplingParam, + "hunyuanvideo-community/HunyuanVideo": + HunyuanSamplingParam, + "Wan-AI/Wan2.1-T2V-1.3B-Diffusers": + WanT2V_1_3B_SamplingParam, + "Wan-AI/Wan2.1-T2V-14B-Diffusers": + WanT2V_14B_SamplingParam, + "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers": + WanI2V_14B_480P_SamplingParam, + "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers": + WanI2V_14B_720P_SamplingParam, + "weizhou03/Wan2.1-Fun-1.3B-InP-Diffusers": + Wan2_1_Fun_1_3B_InP_SamplingParam, + "FastVideo/stepvideo-t2v-diffusers": + StepVideoT2VSamplingParam, + "FastVideo/FastWan2.1-T2V-1.3B-Diffusers": + Wan2_1_Fun_1_3B_InP_SamplingParam, + "FastVideo/FastWan2.1-T2V-14B-480P-Diffusers": + Wan2_1_Fun_1_3B_InP_SamplingParam, # Add other specific weight variants } diff --git a/fastvideo/entrypoints/video_generator.py b/fastvideo/entrypoints/video_generator.py index b17a0460d..dac79a621 100644 --- a/fastvideo/entrypoints/video_generator.py +++ b/fastvideo/entrypoints/video_generator.py @@ -9,8 +9,8 @@ diffusion models. import math import os import time -from typing import Any from copy import deepcopy +from typing import Any import imageio import numpy as np