Skip to content

Commit 5b17ca3

Browse files
Apply suggestions from code review
1 parent 57796da commit 5b17ca3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

optimum/exporters/openvino/convert.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -429,9 +429,9 @@ def ts_patched_forward(*args, **kwargs):
429429

430430
patcher.patched_forward = ts_patched_forward
431431

432-
decoder_kwargs = {}
432+
ts_decoder_kwargs = {}
433433
if library_name == "diffusers" and is_openvino_version(">=", "2025.0"):
434-
decoder_kwargs["trace_kwargs"] = {"check_trace": False}
434+
ts_decoder_kwargs["trace_kwargs"] = {"check_trace": False}
435435

436436
with patcher:
437437
if patch_16bit_model:
@@ -440,9 +440,9 @@ def ts_patched_forward(*args, **kwargs):
440440
__make_16bit_traceable(model)
441441
check_dummy_inputs_are_allowed(model, dummy_inputs)
442442
input_info = _get_input_info(model, config, dummy_inputs)
443-
decoder = TorchScriptPythonDecoder(model, example_input=dummy_inputs, **decoder_kwargs)
443+
ts_decoder = TorchScriptPythonDecoder(model, example_input=dummy_inputs, **ts_decoder_kwargs)
444444
ov_model = convert_model(
445-
decoder,
445+
ts_decoder,
446446
example_input=dummy_inputs,
447447
input=[(item.shape, item.type) for item in input_info],
448448
)

0 commit comments

Comments
 (0)