From adaec5a2e132dc4c86f0f4b8951c912a23d2b452 Mon Sep 17 00:00:00 2001 From: PiotrBLL Date: Tue, 12 Nov 2024 14:53:08 +0100 Subject: [PATCH] Add requirements_hpu.txt --- Dockerfile.hpu | 13 ++++--------- requirements_hpu.txt | 6 ++++++ 2 files changed, 10 insertions(+), 9 deletions(-) create mode 100644 requirements_hpu.txt diff --git a/Dockerfile.hpu b/Dockerfile.hpu index ed2b02e..c654ff4 100644 --- a/Dockerfile.hpu +++ b/Dockerfile.hpu @@ -37,14 +37,9 @@ RUN mkdir -p /usr/local/bin/ffmpeg && \ ADD . /root/whisper WORKDIR /root/whisper -# Copy requirements.txt explicitly -COPY requirements.txt /root/whisper/requirements.txt +# Copy HPU requirements +COPY requirements_hpu.txt /root/whisper/requirements.txt -# Install Python packages from requirements.txt +# Install Python packages RUN pip install --upgrade pip \ - && pip install optimum-habana==1.14.1 \ - && 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 + && pip install -r requirements.txt diff --git a/requirements_hpu.txt b/requirements_hpu.txt new file mode 100644 index 0000000..8f6e6db --- /dev/null +++ b/requirements_hpu.txt @@ -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