# change external model imports as necessary from .smi_ted_light.load import load_smi_ted, Smi_ted, MolTranBertTokenizer from .vq_gan_3d import load_VQGAN, VQGAN