Fix automatic scripting for quantized models (#18089)

2023-06-16 08:51:21 +02:00 · 2023-06-16 08:51:21 +02:00 · 77584d301e
commit 77584d301e
parent 3d79bd1ac5
1 changed files with 2 additions and 0 deletions
--- 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