update ipex api

Signed-off-by: intellinjun <[email protected]>
intel · Jul 3, 2024 · fdd4d34 · fdd4d34
1 parent 86087dc
commit fdd4d34
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/intel_extension_for_transformers/neural_chat/models/model_utils.py b/intel_extension_for_transformers/neural_chat/models/model_utils.py
@@ -840,12 +840,10 @@ def load_model(
 
                     if not use_tpp:
                         try:
-                            model = intel_ipex.optimize(
+                            model = intel_ipex.llm.optimize(
                                 model.eval(),
                                 dtype=torch_dtype,
                                 inplace=True,
-                                level="O1",
-                                auto_kernel_selection=True,
                             )
                         except AssertionError:
                             model = intel_ipex.optimize(