How to use locally downloaded models

#2
by AL58763 - opened

I have downloaded the weights locally on my pc. I want to load the model with these weights. How can i do that?

Apple org

Assuming you have installed the aim package as: pip install git+https://[email protected]/apple/ml-aim.git.

You can load the model with the downloaded checkpoints as follows:

from aim.utils import load_pretrained

model = load_pretrained("aim-600M-2B-imgs", backend="torch", backbone_ckpt_path="<BACKBONE_PATH>",  head_ckpt_path="<HEAD_PATH>")

Please check the repo for more details https://github.com/apple/ml-aim/tree/main.

Thanks @alaaelnouby . I wanted to know since the performance of the aim is scalable, are the larger models being trained currently?

Sign up or log in to comment