Skip to content

Commit fa32df9

Browse files
Pinning vllm for speech-to-text reference (#2273)
Resolving error due to change in vllm main branch.
1 parent c57507b commit fa32df9

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

speech2text/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ RUN pip install --break-system-packages torch==2.7.0 torchaudio==2.7.0 torchvisi
5353
pip install --break-system-packages pandas==2.2.2 toml==0.10.2 unidecode==1.3.8 inflect==7.3.1 librosa==0.10.2 py-libnuma==1.2 numpy==2.0.1 && \
5454
pip install --break-system-packages sox==1.5.0 && \
5555
pip install --break-system-packages setuptools-scm && \
56+
pip install --break-system-packages transformers==4.52.4 && \
5657
pip install --break-system-packages -U openai-whisper
5758

5859
WORKDIR /workspace
@@ -62,7 +63,7 @@ COPY ./ /workspace/
6263
RUN mkdir -p /workspace/third_party && cd /workspace/third_party && \
6364
git clone https://github.com/vllm-project/vllm vllm-cpu && \
6465
cd vllm-cpu && \
65-
git checkout main && \
66+
git checkout b124e1085b1bf977e3dac96d99ffd9d8ddfdb6cc && \
6667
git log -n1 && \
6768
pip3 install --break-system-packages -r requirements/cpu.txt && \
6869
VLLM_TARGET_DEVICE=cpu pip install --break-system-packages . --no-build-isolation

0 commit comments

Comments
 (0)