Skip to content

Commit 5b375f3

Browse files
committed
fix message warning
1 parent eaa068d commit 5b375f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/generation/modeling.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def __init__(
110110
inputs.debugName().split(".")[0] for inputs in model.graph.inputs() if inputs.debugName() != "self"
111111
}
112112
logger.warning(
113-
f"The class `{self.__class__}` has been depreciated for TorchScript model, please use IPEXModel instead"
113+
f"The class `{self.__class__}` has been depreciated for TorchScript model, please use `IPEXModelForCausalLM` instead"
114114
)
115115
else:
116116
self.input_names = set()

0 commit comments

Comments
 (0)