Downloading pre-trained model during docker build

Whenever we call SentenceTransformer(model_name), it downloads the pre-trained model from the server locally. But this happens during the runtime. I want to run this in a docker container and I want to know if there is any command which I can add inside the Dockerfile, so that the pre-trained model gets downloaded during the docker build itself ?

[chanind]

Adding the following into my Dockerfile worked for downloading / setting up the model. It just causes the initial download to happen during the docker build step instead of runtime

RUN python -c 'from sentence_transformers import SentenceTransformer; SentenceTransformer("<model-name>")'

[ahmedshahriar]

Thanks for this @chanind

I added the path param and this saves the pre-trained model in the models directory

RUN python3 -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2'); model.save('models')"