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

Fix MiniCPM-V models running on NPU (#12478)

Browse source
This commit is contained in:
Jin, Qiao 2024-12-02 16:29:46 +08:00 committed by GitHub
parent 54d9a590d4
commit 31c69a8d31
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/src/ipex_llm/transformers/npu_model.py
View file

@ -276,7 +276,8 @@ class _BaseAutoModelClass:
model.share_memory() model.share_memory()
if not pipeline: if not pipeline:
if model.config.model_type in ["qwen2", "llama", "minicpm"]: if (not hasattr(model, 'llm') and
model.config.model_type in ["qwen2", "llama", "minicpm"]):
from ipex_llm.transformers.npu_models.convert import optimize_llm_single_process from ipex_llm.transformers.npu_models.convert import optimize_llm_single_process
optimize_llm_single_process( optimize_llm_single_process(
llm, llm,