Skip to content

Commit f04a0e4

Browse files
apply comments
1 parent 41ef01d commit f04a0e4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/nncf/quantization/algorithms/weight_compression/onnx_backend.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -468,13 +468,15 @@ def _replace_matmul_with_matmulnbits(
468468

469469
class ONNXAWQAlgoAlgoBackend(AWQAlgoBackend, ONNXWeightCompressionAlgoBackend):
470470
@staticmethod
471-
def get_awq_patterns():
471+
def get_awq_patterns() -> dict[str, Callable]:
472472
return get_awq_patterns(
473473
onnx_metatypes.ONNXMatMulMetatype, onnx_metatypes.ONNXMulLayerMetatype, ATOMIC_ACTIVATIONS_OPERATIONS
474474
)
475475

476476
@staticmethod
477-
def scale_insertion_command(source_node, next_nodes, source_node_output_port, scale):
477+
def scale_insertion_command(
478+
source_node: NNCFNode, next_nodes: list[NNCFNode], source_node_output_port: int, scale: np.ndarray
479+
):
478480
return ONNXCommandCreator.multiply_insertion_command(
479481
source_node, next_nodes, source_node_output_port, scale, f"{source_node.node_name}/awq_mul"
480482
)

0 commit comments

Comments
 (0)