diff --git a/Dockerfile_amd b/Dockerfile_amd index 8b7808bea..205e46d93 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=d7e042bf9f7aff10c631212fc71b24895d66eb59 +ENV MOE_KERNELS_BRANCH=v0.8.2 ENV VLLM_TARGET_DEVICE=rocm RUN git clone https://github.com/danieldk/moe-kernels.git && \ cd moe-kernels && \