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

fix: update test models

parent 505b7996
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
tests/unit/encoders/test_clip.py
+ 1
− 1
View file @ b56ed48b
...@@ -5,7 +5,7 @@ from PIL import Image ...@@ -5,7 +5,7 @@ from PIL import Image
from semantic_router.encoders import CLIPEncoder 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) clip_encoder = CLIPEncoder(name=test_model_name)
embed_dim = 64 embed_dim = 64
......
tests/unit/encoders/test_huggingface.py
+ 1
− 1
View file @ b56ed48b
...@@ -5,7 +5,7 @@ import pytest ...@@ -5,7 +5,7 @@ import pytest
from semantic_router.encoders.huggingface import HuggingFaceEncoder 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) encoder = HuggingFaceEncoder(name=test_model_name)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment

:feet::heart: Strive to be the person your dogs believe you are :heart::feet: