Pass pre-commit checks
This commit is contained in:
@@ -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__)
|
||||
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user