Update src/diffusers/pipelines/ltx/pipeline_ltx_condition.py

yiyixuxu · hlky · web-flow · commit a098d94827af · 2025-03-05T21:14:56.000-10:00
Co-authored-by: hlky &lt;hlky@hlky.ac&gt;
diff --git a/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py b/src/diffusers/pipelines/ltx/pipeline_ltx_condition.py
@@ -452,6 +452,7 @@ def _pack_latents(latents: torch.Tensor, patch_size: int = 1, patch_size_t: int
             torch.arange(0, num_frames, patch_size_t, device=device),
             torch.arange(0, height, patch_size, device=device),
             torch.arange(0, width, patch_size, device=device),
+            indexing="ij",
         )
         latent_sample_coords = torch.stack(latent_sample_coords, dim=0)
         latent_coords = latent_sample_coords.unsqueeze(0).repeat(batch_size, 1, 1, 1, 1)

Original file line number	Diff line number	Diff line change
`@@ -452,6 +452,7 @@ def _pack_latents(latents: torch.Tensor, patch_size: int = 1, patch_size_t: int`
`452`	`452`	`torch.arange(0, num_frames, patch_size_t, device=device),`
`453`	`453`	`torch.arange(0, height, patch_size, device=device),`
`454`	`454`	`torch.arange(0, width, patch_size, device=device),`
	`455`	`+ indexing="ij",`
`455`	`456`	`)`
`456`	`457`	`latent_sample_coords = torch.stack(latent_sample_coords, dim=0)`
`457`	`458`	`latent_coords = latent_sample_coords.unsqueeze(0).repeat(batch_size, 1, 1, 1, 1)`