From e342c233fff79a9dfd3adb386340c7edb911b142 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Wed, 15 Apr 2026 11:21:32 +0000
Subject: [PATCH] expirimental fix

---
 Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index f883caf..7b6697e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,8 +9,9 @@ RUN apt-get update && apt-get install -y git \
     libcurand-dev-13-0 \
     libcufft-dev-13-0 \
     libnvjitlink-dev-13-0 && \
-    git clone --depth 1 https://github.com/LMCache/LMCache.git /tmp/lmcache && \
+    git clone https://github.com/neuralwatt/LMCache.git /tmp/lmcache && \
     cd /tmp/lmcache && \
+    git checkout fix/mla-multi-group-kv-cache && \
     CUDA_HOME=/usr/local/cuda \
     TORCH_CUDA_ARCH_LIST="10.0" \
     pip install --no-cache-dir --no-build-isolation . && \