Release CustomModel

Hi, thank you for the great work. May I check if the following models will be release. Thanks again. ```python model_instance = CustomModel(ckpt="models/distil-240709-tot-model_epoch_20.pth") ```