From 8903e92fcdaae3072d081c99ccf56ee56d6b0548 Mon Sep 17 00:00:00 2001 From: gc-fu Date: Fri, 9 Aug 2024 10:11:22 +0800 Subject: [PATCH] Remove duplicate layer --- python/llm/src/ipex_llm/transformers/convert.py | 1 - 1 file changed, 1 deletion(-) diff --git a/python/llm/src/ipex_llm/transformers/convert.py b/python/llm/src/ipex_llm/transformers/convert.py index 3bfba47041b..75c9e9f353a 100644 --- a/python/llm/src/ipex_llm/transformers/convert.py +++ b/python/llm/src/ipex_llm/transformers/convert.py @@ -154,7 +154,6 @@ def is_linear_module(module): VLLM_LINEAR_LIST = [ ColumnParallelLinear, RowParallelLinear, QKVParallelLinear, MergedColumnParallelLinear, - ParallelLMHead ] if 'xpu' in _VLLM_VERSION: VLLM_LINEAR_LIST.append(ParallelLMHead)