diff --git a/Dockerfile_amd b/Dockerfile_amd index 6bfb2bfc..08d69ba0 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -278,7 +278,7 @@ RUN git clone https://github.com/danieldk/marlin-kernels.git && \ FROM kernel-builder AS moe-kernels WORKDIR /usr/src -ENV MOE_KERNELS_BRANCH=127ec5e0fbd2f22fad2d63fbb559d2449e7b5ddb +ENV MOE_KERNELS_BRANCH=b74b163a2e28042068ac8355e07e6dde926a967a ENV VLLM_TARGET_DEVICE=rocm RUN git clone https://github.com/mht-sharma/moe-kernels.git && \ cd moe-kernels && \