fix(ltx2): Fix VAE timing regression for large batch sizes#421

Merged

copybara-service[bot] merged 1 commit into

mainfrom

fix-vae-timing-regression

Jun 17, 2026

Google CLA / cla/google succeeded Jun 16, 2026 in 7s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

✅ 96e9d82 Author: @mbohlool <m***y@google.com>

(Only the first commit for a unique contributor is listed.)

View more details on Google CLA