Skip to content

Commit 62f64c2

Browse files
authored
[deps] Increment Sentence Transformers to v3.3.1 (#464)
* 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
1 parent 08f1a59 commit 62f64c2

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

docker_images/sentence_transformers/Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
FROM tiangolo/uvicorn-gunicorn:python3.8
2-
LABEL maintainer="Omar <omar@huggingface.co>"
1+
FROM tiangolo/uvicorn-gunicorn:python3.10
2+
LABEL maintainer="Tom Aarsen <tom.aarsen@huggingface.co>"
33

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

7-
RUN pip3 install --no-cache-dir torch==1.13.0
7+
RUN pip3 install --no-cache-dir torch==2.4.1
88

99

1010
COPY ./requirements.txt /app
Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
starlette==0.27.0
22
api-inference-community==0.0.32
3-
sentence-transformers==3.0.1
4-
transformers==4.41.1
5-
tokenizers==0.19.1
3+
sentence-transformers==3.3.1
4+
transformers==4.48.0
5+
tokenizers==0.21.0
66
protobuf==3.18.3
7-
huggingface_hub==0.23.3
7+
huggingface_hub==0.27.1
88
sacremoses==0.0.53
99
# dummy.

tests/test_audio.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def test_malformed_audio(self):
3636
def test_dual_channel(self):
3737
bpayload = self.read("sample1_dual.ogg")
3838
payload, params = normalize_payload_audio(bpayload, 16000)
39-
self.assertEqual(payload.shape, (219520,))
39+
self.assertEqual(payload.shape, (219040,))
4040

4141
def test_original_webm(self):
4242
bpayload = self.read("sample1.webm")

0 commit comments

Comments
 (0)