ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Fix import vllm condition (#11682)

Browse source
This commit is contained in:
Guancheng Fu 2024-07-31 13:50:01 +08:00 committed by GitHub
parent 54bf3a23a6
commit afeca38a47
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/src/ipex_llm/transformers/convert.py
View file

@ -73,7 +73,7 @@ def is_vllm_available():
import sys import sys
original_path = sys.path original_path = sys.path
# Temporally remove current directory # Temporally remove current directory
sys.path = [p for p in sys.path if p != ''] sys.path = original_path[1:]
_IS_VLLM_AVAILABLE = importlib.util.find_spec("vllm") is not None _IS_VLLM_AVAILABLE = importlib.util.find_spec("vllm") is not None
sys.path = original_path sys.path = original_path
return _IS_VLLM_AVAILABLE return _IS_VLLM_AVAILABLE