huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

update moe-kernels commit

Browse Source
This commit is contained in:
Mohit Sharma 2024-12-13 17:16:43 +00:00
parent 999eba8096
commit a0c70ca099
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile_amd
View File

@ -270,9 +270,9 @@ RUN python setup.py build
FROM kernel-builder AS moe-kernels FROM kernel-builder AS moe-kernels
WORKDIR /usr/src WORKDIR /usr/src
ENV MOE_KERNELS_BRANCH=b74b163a2e28042068ac8355e07e6dde926a967a ENV MOE_KERNELS_BRANCH=a67b35841774b2056a73806c36661134b5054edd
ENV VLLM_TARGET_DEVICE=rocm ENV VLLM_TARGET_DEVICE=rocm
RUN git clone https://github.com/mht-sharma/moe-kernels.git && \ RUN git clone https://github.com/danieldk/moe-kernels.git && \
cd moe-kernels && \ cd moe-kernels && \
git checkout ${MOE_KERNELS_BRANCH} && \ git checkout ${MOE_KERNELS_BRANCH} && \
python setup.py install python setup.py install