Update latte.py about the xformers input dims Former-commit-id: 421b49c351124170ab476e17cba6e45bd7cf706e