From eaa068d24564651cf14e95ba17220927bd4bea24 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Mon, 4 Mar 2024 10:56:57 +0100 Subject: [PATCH 1/2] Fix warning --- optimum/intel/generation/modeling.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/optimum/intel/generation/modeling.py b/optimum/intel/generation/modeling.py index 0abdafe666..b1ed0502a8 100644 --- a/optimum/intel/generation/modeling.py +++ b/optimum/intel/generation/modeling.py @@ -105,13 +105,13 @@ def __init__( self.normalized_config = NormalizedConfigManager.get_normalized_config_class(config.model_type)(config) self.model_dtype = kwargs.get("model_dtype", None) - logger.warning( - f"The class `{self.__class__}` has been depreciated and will be removed in optimum-intel v1.14, please use IPEXModel instead" - ) if isinstance(model, torch.jit.ScriptModule): self.input_names = { inputs.debugName().split(".")[0] for inputs in model.graph.inputs() if inputs.debugName() != "self" } + logger.warning( + f"The class `{self.__class__}` has been depreciated for TorchScript model, please use IPEXModel instead" + ) else: self.input_names = set() From 5b375f36b2af4a7b15f8b9c2803ac4393aee7174 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Mon, 4 Mar 2024 11:00:21 +0100 Subject: [PATCH 2/2] fix message warning --- optimum/intel/generation/modeling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/intel/generation/modeling.py b/optimum/intel/generation/modeling.py index b1ed0502a8..3d9c657626 100644 --- a/optimum/intel/generation/modeling.py +++ b/optimum/intel/generation/modeling.py @@ -110,7 +110,7 @@ def __init__( inputs.debugName().split(".")[0] for inputs in model.graph.inputs() if inputs.debugName() != "self" } logger.warning( - f"The class `{self.__class__}` has been depreciated for TorchScript model, please use IPEXModel instead" + f"The class `{self.__class__}` has been depreciated for TorchScript model, please use `IPEXModelForCausalLM` instead" ) else: self.input_names = set()