add vae tiling

2026-01-17 07:51:54 +01:00
parent f607112407
commit e4cdbb7eab
6 changed files with 632 additions and 5 deletions
--- a/mlx_video/models/ltx/text_encoder.py
+++ b/mlx_video/models/ltx/text_encoder.py
@@ -918,7 +918,7 @@ class LTX2TextEncoder(nn.Module):
            if response.token == 1 or response.token == 107:  # EOS tokens
                break

-
+        mx.clear_cache()

        # Decode only the new tokens