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