Skip to content

Commit 6eb9569

Browse files
[misc]: re-run yapf on main so pre-commit passes again (#1700)
Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
1 parent c3567eb commit 6eb9569

2 files changed

Lines changed: 6 additions & 13 deletions

File tree

fastvideo/pipelines/basic/minimax_h3/stages/minimax_h3_denoising.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import contextlib
77
from typing import Any
88

9-
109
import torch
1110

1211
from fastvideo.distributed import get_local_torch_device
@@ -106,8 +105,8 @@ def forward(self, batch: ForwardBatch, fastvideo_args: FastVideoArgs) -> Forward
106105
if controller is not None else contextlib.nullcontext())
107106
try:
108107
with denoise_region:
109-
for index, (video_timestep, audio_timestep) in enumerate(zip(video_timesteps, audio_timesteps,
110-
strict=True)):
108+
for index, (video_timestep,
109+
audio_timestep) in enumerate(zip(video_timesteps, audio_timesteps, strict=True)):
111110
unique_timesteps, timestep_indices = row_timestep_plan[index]
112111
# Under torch.compile(mode="reduce-overhead") each denoising
113112
# step must be marked, or cudagraph trees flag cross-step

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)