From 9e3936581a8c1a38f5b96b67aab6a726de1d387d Mon Sep 17 00:00:00 2001 From: PiotrBLL Date: Thu, 7 Nov 2024 18:46:19 +0100 Subject: [PATCH] Add optimum[habana]==1.13.1 in Dockerfile.hpu --- Dockerfile.hpu | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile.hpu b/Dockerfile.hpu index 8c5d0ae..6d4e080 100644 --- a/Dockerfile.hpu +++ b/Dockerfile.hpu @@ -34,4 +34,5 @@ COPY requirements.txt /root/whisper/requirements.txt # Install Python packages from requirements.txt RUN pip install --upgrade pip \ + && pip install optimum[habana]==1.13.1 \ && pip install -r requirements.txt