Consistency decoder train feature?

[dill-shower]

This repository supports the VAE training feature. But is it possible to add support for training LCM as a VAE decoder? This is a recently published feature by openai. This provides much higher

https://github.com/openai/consistencydecoder