Skip to content

Commit dee582d

Browse files
Add trust remote code argument
1 parent 3722624 commit dee582d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

optimum/commands/export/openvino.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -355,7 +355,10 @@ def run(self):
355355
if quantize_after_export:
356356
from optimum.intel import OVModelForCausalLM, OVQuantizer
357357

358-
model = OVModelForCausalLM.from_pretrained(self.args.output)
358+
model = OVModelForCausalLM.from_pretrained(
359+
self.args.output,
360+
trust_remote_code=self.args.trust_remote_code
361+
)
359362
quantizer = OVQuantizer(model)
360363
quantization_config.tokenizer = quantization_config.tokenizer or str(self.args.output)
361364
# TODO: set save_directory=self.args.output once OV is updated to 2024.3

0 commit comments

Comments
 (0)