From 505b7996a9d84cfcebcd2c495542a21af1df52ff Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sat, 1 Jun 2024 16:59:01 +0800
Subject: [PATCH] fix: update model IDs for HF

---
 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 ebbb80e5..a5503e04 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-CLIPModel"
+test_model_name = "hf-internal-testing/tiny-random-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 7c17f98f..07f5aec4 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-bert"
+test_model_name = "hf-internal-testing/tiny-random-bert"
 encoder = HuggingFaceEncoder(name=test_model_name)
 
 
-- 
GitLab