From b56ed48bdb8ccfa39a07b23e7d31beab236d8312 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sat, 1 Jun 2024 17:16:15 +0800
Subject: [PATCH] fix: update test models

---
 tests/unit/encoders/test_clip.py        | 2 +-
 tests/unit/encoders/test_huggingface.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/unit/encoders/test_clip.py b/tests/unit/encoders/test_clip.py
index a5503e04..de997e36 100644
--- a/tests/unit/encoders/test_clip.py
+++ b/tests/unit/encoders/test_clip.py
@@ -5,7 +5,7 @@ from PIL import Image
 
 from semantic_router.encoders import CLIPEncoder
 
-test_model_name = "hf-internal-testing/tiny-random-clip"
+test_model_name = "aurelio-ai/sr-test-clip"
 clip_encoder = CLIPEncoder(name=test_model_name)
 embed_dim = 64
 
diff --git a/tests/unit/encoders/test_huggingface.py b/tests/unit/encoders/test_huggingface.py
index 07f5aec4..f14c7a68 100644
--- a/tests/unit/encoders/test_huggingface.py
+++ b/tests/unit/encoders/test_huggingface.py
@@ -5,7 +5,7 @@ import pytest
 
 from semantic_router.encoders.huggingface import HuggingFaceEncoder
 
-test_model_name = "hf-internal-testing/tiny-random-bert"
+test_model_name = "aurelio-ai/sr-test-huggingface"
 encoder = HuggingFaceEncoder(name=test_model_name)
 
 
-- 
GitLab