Fix latest transformer docker image
Browse filesThe docker image after 4.23.1 seems to contain a broken installation
of PIP, and it's also lacking python3-tk. This commit ensures these
are updated and installed before the rest of the dependencies.
- dockerfile +6 -1
dockerfile
CHANGED
@@ -2,7 +2,12 @@ FROM huggingface/transformers-pytorch-gpu
|
|
2 |
EXPOSE 7860
|
3 |
|
4 |
ADD . /opt/whisper-webui/
|
5 |
-
|
|
|
|
|
|
|
|
|
|
|
6 |
|
7 |
# Note: Models will be downloaded on demand to the directory /root/.cache/whisper.
|
8 |
# You can also bind this directory in the container to somewhere on the host.
|
|
|
2 |
EXPOSE 7860
|
3 |
|
4 |
ADD . /opt/whisper-webui/
|
5 |
+
|
6 |
+
# Latest version of transformers-pytorch-gpu seems to lack tk.
|
7 |
+
# Further, pip install fails, so we must upgrade pip first.
|
8 |
+
RUN apt-get -y install python3-tk
|
9 |
+
RUN python3 -m pip install --upgrade pip &&\
|
10 |
+
python3 -m pip install -r /opt/whisper-webui/requirements.txt
|
11 |
|
12 |
# Note: Models will be downloaded on demand to the directory /root/.cache/whisper.
|
13 |
# You can also bind this directory in the container to somewhere on the host.
|