from .autoencoder_kl_triposg import TripoSGVAEModel