kept latents as jnp

Serenagu525 · Serenagu525 · commit 443243d16408 · 2025-07-20T23:53:44.000Z
diff --git a/src/maxdiffusion/pipelines/ltx_video/ltx_video_pipeline.py b/src/maxdiffusion/pipelines/ltx_video/ltx_video_pipeline.py
@@ -731,7 +731,6 @@ def __call__(
           timestep=noise_cond,
           scheduler_state=scheduler_state,
       )
-    latents = torch.from_numpy(np.array(latents))
     latents = latents[:, num_cond_latents:]
 
     latents = self.patchifier.unpatchify(

Original file line number	Diff line number	Diff line change
`@@ -731,7 +731,6 @@ def __call__(`
`731`	`731`	`timestep=noise_cond,`
`732`	`732`	`scheduler_state=scheduler_state,`
`733`	`733`	`)`
`734`		`- latents = torch.from_numpy(np.array(latents))`
`735`	`734`	`latents = latents[:, num_cond_latents:]`
`736`	`735`
`737`	`736`	`latents = self.patchifier.unpatchify(`