Load TIM model in pretrained_model.py