Skip to content

Commit fbfbb27

Browse files
Update src/nncf/quantization/algorithms/weight_compression/algorithm.py
Co-authored-by: Daniil Lyakhov <[email protected]>
1 parent 96d601c commit fbfbb27

File tree

1 file changed

+3
-4
lines changed
  • src/nncf/quantization/algorithms/weight_compression

1 file changed

+3
-4
lines changed

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

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1099,10 +1099,9 @@ def _get_node_with_weight_from_act_node(
10991099
node_with_weight_edge = graph.get_output_edges_by_port_id(act_node, output_port_id)
11001100
for node_order, edge in enumerate(node_with_weight_edge):
11011101
is_node_with_weights = self._backend_entity.is_node_with_weights(edge.to_node, graph)
1102-
if not is_node_with_weights:
1103-
continue
1104-
node_with_weight = node_with_weight_edge[node_order].to_node
1105-
break
1102+
if is_node_with_weights:
1103+
node_with_weight = node_with_weight_edge[node_order].to_node
1104+
break
11061105
return node_with_weight
11071106

11081107
def _get_weight_port_ids_from_node_with_weight(self, node_with_weight: NNCFNode, graph: NNCFGraph) -> list[int]:

0 commit comments

Comments
 (0)