diff --git a/docker/kimi_k3/kimi_k3_cu12.Dockerfile b/docker/kimi_k3/kimi_k3_cu12.Dockerfile index 5d7a88a9c..16bacb63b 100644 --- a/docker/kimi_k3/kimi_k3_cu12.Dockerfile +++ b/docker/kimi_k3/kimi_k3_cu12.Dockerfile @@ -62,7 +62,7 @@ RUN set -eu; \ # Keep the installed extension modules, but discard Rust and pip build # artifacts that are not used at runtime. RUN rm -rf /sgl-workspace/sglang && \ - git clone --branch kimi-k3 \ + git clone --branch main \ https://github.com/sgl-project/sglang.git /sgl-workspace/sglang && \ cd /sgl-workspace/sglang && \ rm -rf .git && \ diff --git a/docker/kimi_k3/kimi_k3_cu13.Dockerfile b/docker/kimi_k3/kimi_k3_cu13.Dockerfile index 21ca33f90..3fb391504 100644 --- a/docker/kimi_k3/kimi_k3_cu13.Dockerfile +++ b/docker/kimi_k3/kimi_k3_cu13.Dockerfile @@ -53,7 +53,7 @@ ARG TORCH_CUDA_ARCH_LIST="9.0;10.0a;10.3a" # Keep the installed extension modules, but discard Rust and pip build # artifacts that are not used at runtime. RUN rm -rf /sgl-workspace/sglang && \ - git clone --branch kimi-k3 \ + git clone --branch main \ https://github.com/sgl-project/sglang.git /sgl-workspace/sglang && \ cd /sgl-workspace/sglang && \ rm -rf .git && \