external_models/__init__.py

# 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