Skip to content

Commit f64b251

Browse files
committed
fix qwen patch
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent f51777b commit f64b251

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/exporters/ipex/model_patcher.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def _patch_qwen2_model(model):
128128
"""
129129
convert_functions(model, Qwen2Model, "forward", _qwen2_model_forward)
130130
convert_functions(model, Qwen2RMSNorm, "forward", _ipex_rms_layer_norm_forward)
131-
convert_class(model, Qwen2DecoderLayer, _IPEXQwen2DecoderLayer, model.config)
131+
convert_class(model, Qwen2DecoderLayer, _IPEXQwen2DecoderLayer, model.device, model.config)
132132
return model
133133

134134

0 commit comments

Comments
 (0)