diff --git a/requirements/nightly_torch_test.txt b/requirements/nightly_torch_test.txt index a8c22f8e6..72fa13692 100644 --- a/requirements/nightly_torch_test.txt +++ b/requirements/nightly_torch_test.txt @@ -17,7 +17,7 @@ vocos # required for minicpmo_26 test peft pqdm ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests -sentence-transformers # required for embedding tests +sentence-transformers>=5.2.0 # required for embedding tests soundfile # required for audio tests jiwer # required for audio tests timm # required for internvl test diff --git a/requirements/test.in b/requirements/test.in index 2aa36a5d3..2cc3e2f71 100644 --- a/requirements/test.in +++ b/requirements/test.in @@ -19,7 +19,7 @@ vocos # required for minicpmo_26 test peft>=0.15.0 # required for phi-4-mm test pqdm ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests -sentence-transformers # required for embedding tests +sentence-transformers>=5.2.0 # required for embedding tests soundfile # required for audio tests jiwer # required for audio tests tblib # for pickling test exceptions diff --git a/requirements/test.txt b/requirements/test.txt index e734df5e1..4cda9cfad 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -723,7 +723,6 @@ pillow==10.4.0 # perceptron # scikit-image # segmentation-models-pytorch - # sentence-transformers # torchgeo # torchvision platformdirs==4.3.6 @@ -1023,7 +1022,7 @@ segmentation-models-pytorch==0.4.0 # via # terratorch # torchgeo -sentence-transformers==3.2.1 +sentence-transformers==5.2.0 # via # -r requirements/test.in # mteb @@ -1263,6 +1262,7 @@ typing-extensions==4.15.0 # pydantic-core # pydantic-extra-types # pytorch-lightning + # sentence-transformers # sqlalchemy # torch # torchgeo