Skip to content

Commit 702e790

Browse files
committed
fix
1 parent 9fd663e commit 702e790

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/pipelines/pipelines_base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def load_ort_pipeline(
241241

242242
if model is None:
243243
model_id = SUPPORTED_TASKS[targeted_task]["default"]
244-
model = SUPPORTED_TASKS[targeted_task]["class"][0].from_pretrained(model_id)
244+
model = SUPPORTED_TASKS[targeted_task]["class"][0].from_pretrained(model_id, export=True)
245245
elif isinstance(model, str):
246246
model = SUPPORTED_TASKS[targeted_task]["class"][0].from_pretrained(
247247
model, revision=revision, subfolder=subfolder, token=token, **model_kwargs

0 commit comments

Comments
 (0)