Skip to content

Commit e9b48cd

Browse files
committed
removing jax clear cache line
1 parent 6877724 commit e9b48cd

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1639,7 +1639,6 @@ def convert_to_vel(lat, x0, sig):
16391639
graphdef, state = nnx.split(self.transformer)
16401640
state = jax.tree_util.tree_map(lambda x: jax.device_put(x, jax.devices("cpu")[0]), state)
16411641
self.transformer = nnx.merge(graphdef, state)
1642-
jax.clear_caches()
16431642

16441643
vae_start = time.time()
16451644
if getattr(self.vae.config, "timestep_conditioning", False):

0 commit comments

Comments
 (0)