diff --git a/tests/unit/encoders/test_clip.py b/tests/unit/encoders/test_clip.py index ebbb80e50c8833fd5923599a2474b9dcc09f13f0..a5503e04cc8201a8dcec04b9a4d877031632385f 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 7c17f98f4e2cea17205a7543e577cb56b763a1af..07f5aec4114f745f571dc3c0fed82d996bd65a48 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)