From b221f66f84a160abe8b834f25a98959021a82eeb Mon Sep 17 00:00:00 2001 From: Yuhang Yao Date: Fri, 14 Feb 2025 17:03:25 -0800 Subject: [PATCH] Update sglang README.md torch==2.5.1 --- slurm/README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/slurm/README.md b/slurm/README.md index 101064594..662adc14a 100644 --- a/slurm/README.md +++ b/slurm/README.md @@ -5,7 +5,7 @@ conda create -n sglang124 python=3.11 conda activate sglang124 -pip install torch=2.5.1 --index-url https://download.pytorch.org/whl/cu124 +pip install torch==2.5.1 --index-url https://download.pytorch.org/whl/cu124 pip install sgl-kernel --force-reinstall --no-deps pip install "sglang[all]>=0.4.2.post4" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer/ @@ -27,4 +27,4 @@ python scripts/generate_reasoning.py \ --num-generations 2 \ --max-tokens 16384 \ --max-concurrent 200 -``` \ No newline at end of file +```