Skip to content

Commit ad3ec5a

Browse files
committed
pre-commit
1 parent 984f4ba commit ad3ec5a

1 file changed

Lines changed: 4 additions & 10 deletions

File tree

fastvideo/utils.py

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -720,15 +720,11 @@ def maybe_download_model_index(model_name_or_path: str, revision: str | None = N
720720
config_filename = "model_index.json"
721721
try:
722722
filename = f"{subfolder}/{config_filename}" if subfolder else config_filename
723-
model_index_path = hf_hub_download(repo_id=repo_id,
724-
filename=filename,
725-
revision=revision)
723+
model_index_path = hf_hub_download(repo_id=repo_id, filename=filename, revision=revision)
726724
except EntryNotFoundError:
727725
config_filename = "modular_model_index.json"
728726
filename = f"{subfolder}/{config_filename}" if subfolder else config_filename
729-
model_index_path = hf_hub_download(repo_id=repo_id,
730-
filename=filename,
731-
revision=revision)
727+
model_index_path = hf_hub_download(repo_id=repo_id, filename=filename, revision=revision)
732728

733729
# Load the selected manifest.
734730
with open(model_index_path) as f:
@@ -739,14 +735,12 @@ def maybe_download_model_index(model_name_or_path: str, revision: str | None = N
739735
raise ValueError(f"{config_filename} for {model_name_or_path} does not contain _class_name field")
740736

741737
if "_diffusers_version" not in config:
742-
raise ValueError(
743-
f"{config_filename} for {model_name_or_path} does not contain _diffusers_version field")
738+
raise ValueError(f"{config_filename} for {model_name_or_path} does not contain _diffusers_version field")
744739

745740
# Add the pipeline name for downstream use
746741
config["pipeline_name"] = config["_class_name"]
747742

748-
logger.info("Downloaded %s for %s, pipeline: %s", config_filename, model_name_or_path,
749-
config["_class_name"])
743+
logger.info("Downloaded %s for %s, pipeline: %s", config_filename, model_name_or_path, config["_class_name"])
750744
return config
751745

752746
except Exception as e:

0 commit comments

Comments
 (0)