mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-24 16:32:12 +00:00
Fixing dockerfile ?
This commit is contained in:
parent
b77281ac22
commit
f7d47cb35c
@ -98,7 +98,7 @@ ENV HF_HOME=/data \
|
||||
|
||||
|
||||
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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user