Skip to content

Commit 0791c4c

Browse files
committed
fix patching order
1 parent 8405675 commit 0791c4c

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

optimum/exporters/openvino/convert.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -370,11 +370,6 @@ def export_pytorch(
370370
patcher = config.patch_model_for_export(model, model_kwargs=model_kwargs)
371371
patched_forward = patcher.patched_forward
372372

373-
if patch_16bit_model:
374-
from openvino.frontend.pytorch.patch_model import __make_16bit_traceable
375-
376-
__make_16bit_traceable(model)
377-
378373
@functools.wraps(patched_forward)
379374
def ts_patched_forward(*args, **kwargs):
380375
for i in range(len(dict_inputs)):
@@ -388,6 +383,9 @@ def ts_patched_forward(*args, **kwargs):
388383
patcher.patched_forward = ts_patched_forward
389384

390385
with patcher:
386+
if patch_16bit_model:
387+
from openvino.frontend.pytorch.patch_model import __make_16bit_traceable
388+
__make_16bit_traceable(model)
391389
check_dummy_inputs_are_allowed(model, dummy_inputs)
392390
sig = inspect.signature(model.forward) if hasattr(model, "forward") else inspect.signature(model.call)
393391
inputs = config.ordered_inputs(model)

0 commit comments

Comments
 (0)