pass config when convert class

Signed-off-by: jiqing-feng <[email protected]>
huggingface · Dec 18, 2024 · 0c80be3 · 0c80be3
1 parent b0cec9c
commit 0c80be3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/exporters/ipex/model_patcher.py b/optimum/exporters/ipex/model_patcher.py
@@ -119,7 +119,7 @@ def _patch_bert_model(model):
     Patch bert model:
         1. Linear fusion with Linear + Gelu
     """
-    convert_class(model, BertIntermediate, _IPEXIntermediate)
+    convert_class(model, BertIntermediate, _IPEXIntermediate, model.config)
     return model
 
 
@@ -128,7 +128,7 @@ def _patch_vit_model(model):
     Patch vit model:
         1. Linear fusion with Linear + Gelu
     """
-    convert_class(model, ViTIntermediate, _IPEXIntermediate)
+    convert_class(model, ViTIntermediate, _IPEXIntermediate, model.config)
     return model