Skip to content

GitLab

Unverified Commit dd9342e6 authored by Rohan Potdar's avatar Rohan Potdar Committed by GitHub
Browse files

only patch runtime_env for torch >= 2.10 (#38763) 


Signed-off-by: default avatarRohan138 <rohanpotdar138@gmail.com>
parent 8060bb03
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/env_override.py
View file @ dd9342e6
...@@ -500,7 +500,7 @@ if is_torch_equal("2.9.0"): ...@@ -500,7 +500,7 @@ if is_torch_equal("2.9.0"):
# This mirrors the fix in https://github.com/pytorch/pytorch/pull/177558 # This mirrors the fix in https://github.com/pytorch/pytorch/pull/177558
# and can be removed once torch >=2.12 is the minimum supported version. # and can be removed once torch >=2.12 is the minimum supported version.
if not is_torch_equal_or_newer("2.12.0"): if is_torch_equal_or_newer("2.10.0") and not is_torch_equal_or_newer("2.12.0"):
import builtins as _builtins import builtins as _builtins
import pickle import pickle
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment