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/video_vae/init.py
+++ b/mlx_video/models/ltx/video_vae/init.py
@@ -1,3 +1,8 @@
 from mlx_video.models.ltx.video_vae.video_vae import VideoEncoder, VideoDecoder
 from mlx_video.models.ltx.video_vae.encoder import load_vae_encoder, encode_image
 from mlx_video.models.ltx.video_vae.decoder import load_vae_decoder, LTX2VideoDecoder
+from mlx_video.models.ltx.video_vae.tiling import (
+    TilingConfig,
+    SpatialTilingConfig,
+    TemporalTilingConfig,
+)