From 79989c2ac9599e139a473358f5755e16a77e6702 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sat, 1 Jun 2024 18:10:17 +0800
Subject: [PATCH] fix: vit encoder source change

---
 tests/unit/encoders/test_vit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/encoders/test_vit.py b/tests/unit/encoders/test_vit.py
index 9970e310..d7e3d3bf 100644
--- a/tests/unit/encoders/test_vit.py
+++ b/tests/unit/encoders/test_vit.py
@@ -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)
-- 
GitLab