Hi,
Because of some dastardly security block, I’m unable to download a model (specifically distilbert-base-uncased) through my IDE. Specifically, I’m using simpletransformers (built on top of huggingface, or at least uses its models). I tried the from_pretrained method when using huggingface directly, also, but the error is the same:
OSError: Can’t load weights for ‘distilbert-base-uncased’
From where can I download this pretrained model so that I can load it locally?
Thanks very much,
Mark
You can now (since the switch to git models explained here: [Announcement] Model Versioning: Upcoming changes to the model hub) just git clone a model to your laptop and make from_pretrained point to it:
# In a google colab install git-lfs
!sudo apt-get install git-lfs
!git lfs install
# Then
!git clone https://huggingface.co/ORGANIZATION_OR_USER/MODEL_NAME
from transformers import AutoModel
model = AutoModel.from_pretrained('./MODEL_NAME')
For instance:
# In a google colab install git-lfs
!sudo apt-get install git-lfs
!git lfs install
# Then
!git clone https://huggingface.co/facebook/bart-base
from transformers import AutoModel
model = AutoModel.from_pretrained('./bart-base')
cc @julien-c for confirmation
Hi thomwolf,
But how to download only the pytorch model? I found that git cone also downloads tensorflows models, which is useless and time-comsuing for me.
This works for me
from sentence_transformers import SentenceTransformer
# download pretrained
model = SentenceTransformer(‘bert-base-nli-mean-tokens’)
# save to local directory
model.save(“./model/”)
model = None
# load from local file
model = SentenceTransformer(“./model/”)
Didnt work for me with “dandelin/vilt-b32-finetuned-vqa”. Thomwolf’s solution did though.
So I downloaded stable diffusion. I want to add new models, but I don’t know how to download them, at all.
How can I define the revision=float16 here? Because I am trying to download the GPT-J6B model and it is around 24GB. But with revision=float16 I can get it for 12GB.
How to download the 12GB version?