Compare commits
1 Commits
master
...
mla-multi-
| Author | SHA1 | Date | |
|---|---|---|---|
|
e52c6f0a98
|
@@ -13,9 +13,9 @@ RUN apt-get update && apt-get install -y git \
|
|||||||
libcurand-dev-13-0 \
|
libcurand-dev-13-0 \
|
||||||
libcufft-dev-13-0 \
|
libcufft-dev-13-0 \
|
||||||
libnvjitlink-dev-13-0 && \
|
libnvjitlink-dev-13-0 && \
|
||||||
git clone https://github.com/biondizzle/LMCache.git /tmp/lmcache && \
|
git clone https://github.com/Byteflux/LMCache.git /tmp/lmcache && \
|
||||||
cd /tmp/lmcache && \
|
cd /tmp/lmcache && \
|
||||||
git checkout feat/redis-ttl && \
|
git checkout mla-multi-group-kv-cache-with-redis && \
|
||||||
CUDA_HOME=/usr/local/cuda \
|
CUDA_HOME=/usr/local/cuda \
|
||||||
TORCH_CUDA_ARCH_LIST="10.0" \
|
TORCH_CUDA_ARCH_LIST="10.0" \
|
||||||
pip install --no-cache-dir --no-build-isolation . && \
|
pip install --no-cache-dir --no-build-isolation . && \
|
||||||
|
|||||||
Reference in New Issue
Block a user