diff --git a/src/bindings/python/src/openvino/frontend/pytorch/decoder.py b/src/bindings/python/src/openvino/frontend/pytorch/decoder.py index b904c23650c..e7be801ba33 100644 --- a/src/bindings/python/src/openvino/frontend/pytorch/decoder.py +++ b/src/bindings/python/src/openvino/frontend/pytorch/decoder.py @@ -176,6 +176,8 @@ class TorchScriptPythonDecoder (Decoder): break if not skip_freeze: f_model = torch.jit.freeze(scripted) + else: + f_model = scripted else: f_model = pt_module