From ec53a4dce76216694cafeaeb74987c007d5f933c Mon Sep 17 00:00:00 2001 From: JerryZhou54 Date: Tue, 15 Apr 2025 19:47:35 +0000 Subject: [PATCH] Pass pre-commit checks --- fastvideo/v1/pipelines/__init__.py | 4 +--- fastvideo/v1/utils.py | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/fastvideo/v1/pipelines/__init__.py b/fastvideo/v1/pipelines/__init__.py index 391620eff..0b6717aac 100644 --- a/fastvideo/v1/pipelines/__init__.py +++ b/fastvideo/v1/pipelines/__init__.py @@ -11,9 +11,7 @@ from fastvideo.v1.pipelines.composed_pipeline_base import ComposedPipelineBase from fastvideo.v1.pipelines.pipeline_batch_info import ForwardBatch from fastvideo.v1.pipelines.pipeline_registry import PipelineRegistry from fastvideo.v1.utils import (maybe_download_model, - verify_model_config_and_directory, diff_keys, - update_in_place) -from fastvideo.v1.configs.registry import get_pipeline_config_for_name + verify_model_config_and_directory) logger = init_logger(__name__) diff --git a/fastvideo/v1/utils.py b/fastvideo/v1/utils.py index 109342e36..9df0816fa 100644 --- a/fastvideo/v1/utils.py +++ b/fastvideo/v1/utils.py @@ -489,7 +489,7 @@ def maybe_download_model_index(model_name_or_path: str) -> Dict[str, Any]: # Load the model_index.json with open(model_index_path) as f: - config = json.load(f) + config: Dict[str, Any] = json.load(f) # Verify it has the required fields if "_class_name" not in config: