loading the lite/tiny version of MedSam

[mohammadj96]

Hello, I want to load and check the performance of the Lite or the Tiny version of MedSam. I've followed the code named MedSAM_Inference.py. But there is a line where you should load the model (line 110: medsam_model = sam_model_registry"vit_b") and when I change that to my chosen model (for example, /work_dir/MedSAM/lite_medsam.pth), it raises an error. I've tried vit-b, vit-h, and vit-l options, but all of them gave me the same error. Is there a tutorial where I can find a guide for loading these two models (Lite/Tiny)?