File tree 1 file changed +4
-4
lines changed
optimum/exporters/openvino
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -429,9 +429,9 @@ def ts_patched_forward(*args, **kwargs):
429
429
430
430
patcher .patched_forward = ts_patched_forward
431
431
432
- decoder_kwargs = {}
432
+ ts_decoder_kwargs = {}
433
433
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 }
435
435
436
436
with patcher :
437
437
if patch_16bit_model :
@@ -440,9 +440,9 @@ def ts_patched_forward(*args, **kwargs):
440
440
__make_16bit_traceable (model )
441
441
check_dummy_inputs_are_allowed (model , dummy_inputs )
442
442
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 )
444
444
ov_model = convert_model (
445
- decoder ,
445
+ ts_decoder ,
446
446
example_input = dummy_inputs ,
447
447
input = [(item .shape , item .type ) for item in input_info ],
448
448
)
You can’t perform that action at this time.
0 commit comments