diff --git a/fastvideo/configs/sample/registry.py b/fastvideo/configs/sample/registry.py index f54062dfa..beaceeea1 100644 --- a/fastvideo/configs/sample/registry.py +++ b/fastvideo/configs/sample/registry.py @@ -6,7 +6,7 @@ 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, @@ -28,7 +28,6 @@ SAMPLING_PARAM_REGISTRY: dict[str, Any] = { "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": FastWanT2V480PConfig, "Wan-AI/Wan2.2-TI2V-5B-Diffusers": Wan2_2_TI2V_5B_SamplingParam, "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, 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