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

fix: vit encoder source change

parent a0300f07
No related branches found
No related tags found
No related merge requests found
......@@ -6,7 +6,7 @@ from PIL import Image
from semantic_router.encoders import VitEncoder
from transformers import AutoTokenizer, AutoModel
test_model_name = "hf-internal-testing/tiny-random-vit"
test_model_name = "aurelio-ai/sr-test-vit"
# force the model download
tokenizer = AutoTokenizer.from_pretrained(test_model_name, force_download=True)
......
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