diff --git a/Dockerfile_amd b/Dockerfile_amd index 10001012..201d9010 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -279,7 +279,7 @@ RUN git clone https://github.com/danieldk/marlin-kernels.git && \ FROM kernel-builder AS moe-kernels WORKDIR /usr/src -ENV MOE_KERNELS_BRANCH=0c8f8ed941635025e277d6dc6f7324827855cc40 +ENV MOE_KERNELS_BRANCH=9fb72ccb00b8da96a41f9b9c48f305c7dbd266bc ENV VLLM_TARGET_DEVICE=rocm RUN git clone https://github.com/mht-sharma/moe-kernels.git && \ cd moe-kernels && \