diff --git a/optimum/intel/generation/modeling.py b/optimum/intel/generation/modeling.py index 0abdafe666..3d9c657626 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 `IPEXModelForCausalLM` instead" + ) else: self.input_names = set()