mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-25 03:52:08 +00:00
Fixing dockerfile ?
This commit is contained in:
parent
b77281ac22
commit
f7d47cb35c
@ -98,7 +98,7 @@ ENV HF_HOME=/data \
|
|||||||
|
|
||||||
|
|
||||||
WORKDIR /usr/src
|
WORKDIR /usr/src
|
||||||
RUN pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/test/xpu
|
RUN pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/test/xpu
|
||||||
|
|
||||||
RUN pip install triton-xpu==3.2.0b1 --no-cache-dir
|
RUN pip install triton-xpu==3.2.0b1 --no-cache-dir
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user