From 907896d533ae7089c30cd98790975c4ad5dd6b48 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 22 Dec 2025 13:41:41 +0530 Subject: [PATCH] simplify and clean up --- scripts/test_ltx2_audio_conversion.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/test_ltx2_audio_conversion.py b/scripts/test_ltx2_audio_conversion.py index 251d0b64e9..649b6d06d6 100644 --- a/scripts/test_ltx2_audio_conversion.py +++ b/scripts/test_ltx2_audio_conversion.py @@ -91,7 +91,9 @@ def main() -> None: levels = len(diffusers_model.decoder.channel_multipliers) latent_size = diffusers_model.decoder.resolution // (2 ** (levels - 1)) - dummy = torch.randn(args.batch, diffusers_model.decoder.latent_channels, latent_size, latent_size, device=device, dtype=dtype) + dummy = torch.randn( + args.batch, diffusers_model.decoder.latent_channels, latent_size, latent_size, device=device, dtype=dtype + ) with torch.no_grad(): original_out = original_decoder(dummy)