File tree 2 files changed +3
-1
lines changed
optimum/exporters/openvino
2 files changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -323,6 +323,7 @@ class StoreAttr(object):
323
323
fn_get_submodels = fn_get_submodels ,
324
324
preprocessors = preprocessors ,
325
325
device = device ,
326
+ trust_remote_code = trust_remote_code ,
326
327
** kwargs_shapes ,
327
328
)
328
329
Original file line number Diff line number Diff line change @@ -493,6 +493,7 @@ def export_from_model(
493
493
fn_get_submodels : Optional [Callable ] = None ,
494
494
preprocessors : List = None ,
495
495
device : str = "cpu" ,
496
+ trust_remote_code : bool = False ,
496
497
** kwargs_shapes ,
497
498
):
498
499
if ov_config is not None and ov_config .quantization_config and not is_nncf_available ():
@@ -605,7 +606,7 @@ def export_from_model(
605
606
generation_config .save_pretrained (output )
606
607
607
608
model_name_or_path = model .config ._name_or_path
608
- maybe_save_preprocessors (model_name_or_path , output )
609
+ maybe_save_preprocessors (model_name_or_path , output , trust_remote_code = trust_remote_code )
609
610
610
611
files_subpaths = ["openvino_" + model_name + ".xml" for model_name in models_and_export_configs .keys ()]
611
612
You can’t perform that action at this time.
0 commit comments