Skip to content

Commit

Permalink
[deps] Increment Sentence Transformers to v3.3.1 (#464)
Browse files Browse the repository at this point in the history
* Increment Sentence Transformers to v3.3.1

* Upgrade Docker Python version; take maintainership

and increase torch version

* Update the expected payload shape to match other tests
  • Loading branch information
tomaarsen authored Jan 17, 2025
1 parent 08f1a59 commit 62f64c2
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 8 deletions.
6 changes: 3 additions & 3 deletions docker_images/sentence_transformers/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
FROM tiangolo/uvicorn-gunicorn:python3.8
LABEL maintainer="Omar <omar@huggingface.co>"
FROM tiangolo/uvicorn-gunicorn:python3.10
LABEL maintainer="Tom Aarsen <tom.aarsen@huggingface.co>"

# Add any system dependency here
# RUN apt-get update -y && apt-get install libXXX -y

RUN pip3 install --no-cache-dir torch==1.13.0
RUN pip3 install --no-cache-dir torch==2.4.1


COPY ./requirements.txt /app
Expand Down
8 changes: 4 additions & 4 deletions docker_images/sentence_transformers/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
starlette==0.27.0
api-inference-community==0.0.32
sentence-transformers==3.0.1
transformers==4.41.1
tokenizers==0.19.1
sentence-transformers==3.3.1
transformers==4.48.0
tokenizers==0.21.0
protobuf==3.18.3
huggingface_hub==0.23.3
huggingface_hub==0.27.1
sacremoses==0.0.53
# dummy.
2 changes: 1 addition & 1 deletion tests/test_audio.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ def test_malformed_audio(self):
def test_dual_channel(self):
bpayload = self.read("sample1_dual.ogg")
payload, params = normalize_payload_audio(bpayload, 16000)
self.assertEqual(payload.shape, (219520,))
self.assertEqual(payload.shape, (219040,))

def test_original_webm(self):
bpayload = self.read("sample1.webm")
Expand Down

0 comments on commit 62f64c2

Please sign in to comment.