Automated PR - 2026-01-08

This commit is contained in:
sync-bot
2026-01-08 15:29:32 +00:00
parent 7179f0d0e3
commit ac560b4775
7 changed files with 106 additions and 68 deletions
@@ -68,7 +68,10 @@ class LatentsDecoder:
self.vae = load_video_vae_decoder(model_path, device=self.device, dtype=torch.bfloat16)
if vae_tiling:
self.vae.enable_tiling()
logger.warning(
"VAE tiling is not yet implemented in this script. "
"Continuing without tiling - this may cause OOM errors for large resolutions."
)
if with_audio:
with console.status(f"[bold]Loading audio VAE decoder from {model_path}...", spinner="dots"):
@@ -48,6 +48,13 @@ def preprocess_dataset( # noqa: PLR0913
with_audio: bool = False,
) -> None:
"""Run the preprocessing pipeline with the given arguments."""
# VAE tiling is not yet implemented
if vae_tiling:
logger.warning(
"VAE tiling is not yet implemented in this script. "
"Continuing without tiling - this may cause OOM errors for large resolutions."
)
# Validate dataset file
_validate_dataset_file(dataset_file)
@@ -489,7 +489,10 @@ def compute_latents( # noqa: PLR0913, PLR0915
vae = load_video_vae_encoder(model_path, device=torch_device, dtype=torch.bfloat16)
if vae_tiling:
vae.enable_tiling()
logger.warning(
"VAE tiling is not yet implemented in this script. "
"Continuing without tiling - this may cause OOM errors for large resolutions."
)
# Load audio VAE encoder and audio processor if needed
audio_vae_encoder = None