chore: upgrade torch 2.8.0 (#8836)
This commit is contained in:
2
.github/workflows/vllm-dependency-test.yml
vendored
2
.github/workflows/vllm-dependency-test.yml
vendored
@@ -29,9 +29,9 @@ jobs:
|
|||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
bash scripts/ci_install_dependency.sh
|
|
||||||
pip install "vllm==0.10.0"
|
pip install "vllm==0.10.0"
|
||||||
pip install "bitsandbytes>=0.44.0"
|
pip install "bitsandbytes>=0.44.0"
|
||||||
|
bash scripts/ci_install_dependency.sh
|
||||||
|
|
||||||
- name: Run VLLM dependency tests
|
- name: Run VLLM dependency tests
|
||||||
timeout-minutes: 60
|
timeout-minutes: 60
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ name = "sglang"
|
|||||||
version = "0.4.10.post2"
|
version = "0.4.10.post2"
|
||||||
description = "SGLang is yet another fast serving framework for large language models and vision language models."
|
description = "SGLang is yet another fast serving framework for large language models and vision language models."
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
requires-python = ">=3.8"
|
requires-python = ">=3.9"
|
||||||
license = { file = "LICENSE" }
|
license = { file = "LICENSE" }
|
||||||
classifiers = [
|
classifiers = [
|
||||||
"Programming Language :: Python :: 3",
|
"Programming Language :: Python :: 3",
|
||||||
@@ -54,10 +54,10 @@ runtime_common = [
|
|||||||
|
|
||||||
srt = [
|
srt = [
|
||||||
"sglang[runtime_common]",
|
"sglang[runtime_common]",
|
||||||
"sgl-kernel==0.2.8",
|
"sgl-kernel==0.3.2",
|
||||||
"torch==2.7.1",
|
"torch==2.8.0",
|
||||||
"torchaudio==2.7.1",
|
"torchaudio==2.8.0",
|
||||||
"torchvision==0.22.1",
|
"torchvision",
|
||||||
"cuda-python",
|
"cuda-python",
|
||||||
"einops",
|
"einops",
|
||||||
"flashinfer_python==0.2.10",
|
"flashinfer_python==0.2.10",
|
||||||
@@ -66,9 +66,9 @@ srt = [
|
|||||||
blackwell = [
|
blackwell = [
|
||||||
"sglang[runtime_common]",
|
"sglang[runtime_common]",
|
||||||
"sgl-kernel",
|
"sgl-kernel",
|
||||||
"torch==2.7.1",
|
"torch==2.8.0",
|
||||||
"torchaudio==2.7.1",
|
"torchaudio==2.8.0",
|
||||||
"torchvision==0.22.1",
|
"torchvision",
|
||||||
"cuda-python",
|
"cuda-python",
|
||||||
"einops",
|
"einops",
|
||||||
"flashinfer_python==0.2.10",
|
"flashinfer_python==0.2.10",
|
||||||
|
|||||||
@@ -649,7 +649,7 @@ def _set_envs_and_config(server_args: ServerArgs):
|
|||||||
if _is_cuda:
|
if _is_cuda:
|
||||||
assert_pkg_version(
|
assert_pkg_version(
|
||||||
"sgl-kernel",
|
"sgl-kernel",
|
||||||
"0.2.8",
|
"0.3.2",
|
||||||
"Please reinstall the latest version with `pip install sgl-kernel --force-reinstall`",
|
"Please reinstall the latest version with `pip install sgl-kernel --force-reinstall`",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ rm -rf /usr/local/lib/python3.10/dist-packages/flashinfer*
|
|||||||
rm -rf /usr/local/lib/python3.10/dist-packages/sgl_kernel*
|
rm -rf /usr/local/lib/python3.10/dist-packages/sgl_kernel*
|
||||||
|
|
||||||
# Install the main package
|
# Install the main package
|
||||||
pip install -e "python[dev]"
|
pip install -e "python[dev]" --extra-index-url https://download.pytorch.org/whl/test/cu126
|
||||||
|
|
||||||
# Show current packages
|
# Show current packages
|
||||||
pip list
|
pip list
|
||||||
|
|||||||
Reference in New Issue
Block a user