From 489e5b0fbe6749eac4314604d07258f81390a720 Mon Sep 17 00:00:00 2001 From: "Wang, Yi A" Date: Mon, 28 Oct 2024 01:38:06 -0700 Subject: [PATCH] add xpu triton in dockerfile, or will show "Could not import Flash Attention enabled models: No module named 'triton'" Signed-off-by: Wang, Yi A --- Dockerfile_intel | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile_intel b/Dockerfile_intel index 96f24248..f9b1cd13 100644 --- a/Dockerfile_intel +++ b/Dockerfile_intel @@ -93,6 +93,7 @@ ENV HF_HOME=/data \ WORKDIR /usr/src RUN pip install torch==2.3.1+cxx11.abi torchvision==0.18.1+cxx11.abi torchaudio==2.3.1+cxx11.abi intel-extension-for-pytorch==2.3.110+xpu oneccl_bind_pt==2.3.100+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ --no-cache-dir +RUN pip install triton-xpu==3.0.0b2 --no-cache-dir # Install server COPY proto proto