From 77584d301e52041a1095ee42537b4280c3a996c6 Mon Sep 17 00:00:00 2001 From: Maxim Vafin Date: Fri, 16 Jun 2023 08:51:21 +0200 Subject: [PATCH] Fix automatic scripting for quantized models (#18089) --- src/bindings/python/src/openvino/frontend/pytorch/decoder.py | 2 ++ 1 file changed, 2 insertions(+) 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