nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[misc]: re-run yapf on main so pre-commit passes again (#1700)

Browse Source 
Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
KyleNeverGivesUp
2026-08-07 16:17:52 -07:00
committed by GitHub
co-authored by Claude Opus 5
parent c3567eb468
commit 6eb95693a1
2 changed files with 6 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastvideo/pipelines/basic/minimax_h3/stages/minimax_h3_denoising.py
+2 -3
View File
@@ -6,7 +6,6 @@ from __future__ import annotations
import contextlib
from typing import Any
import torch
from fastvideo.distributed import get_local_torch_device
@@ -106,8 +105,8 @@ class MiniMaxH3DenoisingStage(PipelineStage):
if controller is not None else contextlib.nullcontext())
try:
with denoise_region:
for index, (video_timestep, audio_timestep) in enumerate(zip(video_timesteps, audio_timesteps,
strict=True)):
for index, (video_timestep,
audio_timestep) in enumerate(zip(video_timesteps, audio_timesteps, strict=True)):
unique_timesteps, timestep_indices = row_timestep_plan[index]
# Under torch.compile(mode="reduce-overhead") each denoising
# step must be marked, or cudagraph trees flag cross-step
fastvideo/utils.py
+4 -10
View File
@@ -720,15 +720,11 @@ def maybe_download_model_index(model_name_or_path: str, revision: str | None = N
config_filename = "model_index.json"
try:
filename = f"{subfolder}/{config_filename}" if subfolder else config_filename
model_index_path = hf_hub_download(repo_id=repo_id,
filename=filename,
revision=revision)
model_index_path = hf_hub_download(repo_id=repo_id, filename=filename, revision=revision)
except EntryNotFoundError:
config_filename = "modular_model_index.json"
filename = f"{subfolder}/{config_filename}" if subfolder else config_filename
model_index_path = hf_hub_download(repo_id=repo_id,
filename=filename,
revision=revision)
model_index_path = hf_hub_download(repo_id=repo_id, filename=filename, revision=revision)
# Load the selected manifest.
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
raise ValueError(f"{config_filename} for {model_name_or_path} does not contain _class_name field")
if "_diffusers_version" not in config:
raise ValueError(
f"{config_filename} for {model_name_or_path} does not contain _diffusers_version field")
raise ValueError(f"{config_filename} for {model_name_or_path} does not contain _diffusers_version field")
# Add the pipeline name for downstream use
config["pipeline_name"] = config["_class_name"]
logger.info("Downloaded %s for %s, pipeline: %s", config_filename, model_name_or_path,
config["_class_name"])
logger.info("Downloaded %s for %s, pipeline: %s", config_filename, model_name_or_path, config["_class_name"])
return config
except Exception as e: