Skip to content

Commit 35aa8af

Browse files
minor update
1 parent c2e41a9 commit 35aa8af

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

nncf/onnx/graph/model_transformer.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,8 @@ class ONNXModelTransformer(ModelTransformer):
4747
ZERO_POINT_NAME_PREFIX = "zero_point_"
4848

4949
def __init__(self, model: onnx.ModelProto):
50-
inferred_model = onnx.shape_inference.infer_shapes(model)
51-
super().__init__(inferred_model)
52-
self.onnx_model_extractor = onnx.utils.Extractor(inferred_model)
50+
super().__init__(model)
51+
self.onnx_model_extractor = onnx.utils.Extractor(model)
5352

5453
def _get_target_edge(
5554
self,

0 commit comments

Comments
 (0)