Skip to content

Commit eaa068d

Browse files
committed
Fix warning
1 parent 99cf01f commit eaa068d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

optimum/intel/generation/modeling.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -105,13 +105,13 @@ def __init__(
105105
self.normalized_config = NormalizedConfigManager.get_normalized_config_class(config.model_type)(config)
106106
self.model_dtype = kwargs.get("model_dtype", None)
107107

108-
logger.warning(
109-
f"The class `{self.__class__}` has been depreciated and will be removed in optimum-intel v1.14, please use IPEXModel instead"
110-
)
111108
if isinstance(model, torch.jit.ScriptModule):
112109
self.input_names = {
113110
inputs.debugName().split(".")[0] for inputs in model.graph.inputs() if inputs.debugName() != "self"
114111
}
112+
logger.warning(
113+
f"The class `{self.__class__}` has been depreciated for TorchScript model, please use IPEXModel instead"
114+
)
115115
else:
116116
self.input_names = set()
117117

0 commit comments

Comments
 (0)