Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 06:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add requirements_hpu.txt

Browse Source
This commit is contained in:
PiotrBLL 2024-11-12 14:53:08 +01:00
parent e1545f4776
commit adaec5a2e1
2 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
Dockerfile.hpu
View File

@ -37,14 +37,9 @@ RUN mkdir -p /usr/local/bin/ffmpeg && \
ADD . /root/whisper ADD . /root/whisper
WORKDIR /root/whisper WORKDIR /root/whisper
# Copy requirements.txt explicitly # Copy HPU requirements
COPY requirements.txt /root/whisper/requirements.txt COPY requirements_hpu.txt /root/whisper/requirements.txt
# Install Python packages from requirements.txt # Install Python packages
RUN pip install --upgrade pip \ RUN pip install --upgrade pip \
&& pip install optimum-habana==1.14.1 \ && pip install -r requirements.txt
&& pip install transformers==4.33.0 \
&& pip install huggingface-hub==0.26.2 \
&& pip install tiktoken==0.8.0 \
&& pip install torch-geometric==2.6.1 \
&& pip install numba==0.60.0

6
requirements_hpu.txt Normal file
View File

@ -0,0 +1,6 @@
optimum-habana==1.14.1
transformers==4.45.2
huggingface-hub==0.26.2
tiktoken==0.8.0
torch-geometric==2.6.1
numba==0.60.0