We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent abe8704 commit aa4d4e6Copy full SHA for aa4d4e6
optimum/intel/pipelines/pipeline_base.py
@@ -113,8 +113,7 @@ def load_ipex_model(
113
model_id = model
114
try:
115
config = AutoConfig.from_pretrained(model)
116
- torchscript = getattr(config, "torchscript", None)
117
- export = False if torchscript else True
+ export = not getattr(config, "torchscript", False)
118
except RuntimeError:
119
logger.warning("We will use IPEXModel with export=True to export the model")
120
export = True
0 commit comments