AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Add option to enable VAE tiling in upsampling test script

Browse Source
This commit is contained in:
Daniel Gu
2026-01-06 08:07:33 +01:00
parent a7d6916afc
commit 245d056c7d
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/ltx2_test_latent_upsampler.py
View File

@@ -52,6 +52,7 @@ def parse_args():
parser.add_argument("--device", type=str, default="cuda:0")
parser.add_argument("--dtype", type=str, default="bf16")
parser.add_argument("--cpu_offload", action="store_true")
parser.add_argument("--vae_tiling", action="store_true")
parser.add_argument(
"--output_dir",
@@ -120,6 +121,8 @@ def main(args):
)
upsample_pipeline = LTX2LatentUpsamplePipeline(vae=vae, latent_upsampler=latent_upsampler)
upsample_pipeline.to(device=args.device)
if args.vae_tiling:
upsample_pipeline.enable_vae_tiling()
video = upsample_pipeline(
video=video,