Skip to content

Commit 9d5da2d

Browse files
[OpenVINOQuantizer] return annotated model (#3263)
### Changes Mode return is added to `annotate` method of the OpenVINOQuantizer ### Reason for changes To align with the signature
1 parent 66ee9da commit 9d5da2d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

nncf/experimental/torch/fx/quantization/quantizer/openvino_quantizer.py

+1
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,7 @@ def annotate(self, model: torch.fx.GraphModule) -> torch.fx.GraphModule:
154154
for node, annotation in node_vs_torch_annotation.items():
155155
assert QUANT_ANNOTATION_KEY not in node.meta
156156
node.meta[QUANT_ANNOTATION_KEY] = annotation
157+
return model
157158

158159
@staticmethod
159160
def _get_unified_scales_root_quantizer_id(

0 commit comments

Comments
 (0)