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

simplify and clean up

Browse Source
This commit is contained in:
sayakpaul
2025-12-22 13:41:41 +05:30
parent e54cd6bb1d
commit 907896d533
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/test_ltx2_audio_conversion.py
View File

@@ -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)