From c46c46ef77975e07ecd8cf94990a9c9621d1b1e7 Mon Sep 17 00:00:00 2001 From: Xinyu Dong Date: Wed, 31 Dec 2025 13:17:06 +0800 Subject: [PATCH] [Docs] Update torch and ops for mimo v2 --- docs/source/installation.md | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/docs/source/installation.md b/docs/source/installation.md index d1f4115..f96ee56 100644 --- a/docs/source/installation.md +++ b/docs/source/installation.md @@ -76,11 +76,19 @@ cp vllm_kunlun/patches/eval_frame.py /root/miniconda/envs/vllm_kunlun_0.10.1.1/l wget -O xpytorch-cp310-torch251-ubuntu2004-x64.run https://baidu-kunlun-public.su.bcebos.com/v1/baidu-kunlun-share/1130/xpytorch-cp310-torch251-ubuntu2004-x64.run?authorization=bce-auth-v1%2FALTAKypXxBzU7gg4Mk4K4c6OYR%2F2025-12-02T05%3A01%3A27Z%2F-1%2Fhost%2Ff3cf499234f82303891aed2bcb0628918e379a21e841a3fac6bd94afef491ff7 bash xpytorch-cp310-torch251-ubuntu2004-x64.run ``` +## Install the KL3-customized build of PyTorch(Only MIMO V2) +``` +wget -O xpytorch-cp310-torch251-ubuntu2004-x64.run https://klx-sdk-release-public.su.bcebos.com/kunlun2aiak_output/1231/xpytorch-cp310-torch251-ubuntu2004-x64.run +``` ## Install custom ops ``` pip install "https://baidu-kunlun-public.su.bcebos.com/v1/baidu-kunlun-share/1130/xtorch_ops-0.1.2209%2B6752ad20-cp310-cp310-linux_x86_64.whl?authorization=bce-auth-v1%2FALTAKypXxBzU7gg4Mk4K4c6OYR%2F2025-12-05T06%3A18%3A00Z%2F-1%2Fhost%2F14936c2b7e7c557c1400e4c467c79f7a9217374a7aa4a046711ac4d948f460cd" ``` +## Install custom ops(Only MIMO V2) +``` +pip install "https://vllm-ai-models.bj.bcebos.com/v1/vLLM-Kunlun/ops/swa/xtorch_ops-0.1.2109%252B523cb26d-cp310-cp310-linux_x86_64.whl" +``` ## Install the KLX3 custom Triton build ```