Skip to content

Commit aa4d4e6

Browse files
Update optimum/intel/pipelines/pipeline_base.py
Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
1 parent abe8704 commit aa4d4e6

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

optimum/intel/pipelines/pipeline_base.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -113,8 +113,7 @@ def load_ipex_model(
113113
model_id = model
114114
try:
115115
config = AutoConfig.from_pretrained(model)
116-
torchscript = getattr(config, "torchscript", None)
117-
export = False if torchscript else True
116+
export = not getattr(config, "torchscript", False)
118117
except RuntimeError:
119118
logger.warning("We will use IPEXModel with export=True to export the model")
120119
export = True

0 commit comments

Comments
 (0)