Skip to content
Snippets Groups Projects
Unverified Commit a0300f07 authored by James Briggs's avatar James Briggs
Browse files

fix: force download of ViT encoder

parent b0eb587b
No related branches found
No related tags found
No related merge requests found
...@@ -4,8 +4,14 @@ import torch ...@@ -4,8 +4,14 @@ import torch
from PIL import Image from PIL import Image
from semantic_router.encoders import VitEncoder from semantic_router.encoders import VitEncoder
from transformers import AutoTokenizer, AutoModel
test_model_name = "hf-internal-testing/tiny-random-vit" test_model_name = "hf-internal-testing/tiny-random-vit"
# force the model download
tokenizer = AutoTokenizer.from_pretrained(test_model_name, force_download=True)
model = AutoModel.from_pretrained(test_model_name, force_download=True)
vit_encoder = VitEncoder(name=test_model_name) vit_encoder = VitEncoder(name=test_model_name)
embed_dim = 32 embed_dim = 32
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment