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

LLM: update check for deepspeed (#9838)

Browse source
This commit is contained in:
Ruonan Wang 2024-01-05 16:44:10 +08:00 committed by GitHub
parent f25d23dfbf
commit a60bda3324
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
python/llm/src/bigdl/llm/transformers/convert.py
View file

@ -59,7 +59,16 @@ def is_auto_awq_available():
def is_deepspeed_available(): def is_deepspeed_available():
return importlib.util.find_spec("deepspeed") is not None spec = importlib.util.find_spec("deepspeed")
if spec is not None:
deepspeed_path = spec.submodule_search_locations[0]
if deepspeed_path != os.path.join(os.getcwd(), "deepspeed"):
return True
else:
# not deepspeed package, just local dir
return False
else:
return False
if is_auto_gptq_available(): if is_auto_gptq_available():