diff --git a/onnxslim/core/optimization/dead_node_elimination.py b/onnxslim/core/optimization/dead_node_elimination.py index 94d35dd..f2be9e2 100644 --- a/onnxslim/core/optimization/dead_node_elimination.py +++ b/onnxslim/core/optimization/dead_node_elimination.py @@ -58,7 +58,7 @@ def dead_node_elimination(graph, is_subgraph=False): isinstance(node.inputs[0], Constant) and isinstance(node.inputs[1], Variable) ): idx, constant_variable = get_constant_variable(node, return_idx=True) - if np.all(constant_variable.values == 1):git + if np.all(constant_variable.values == 1): var_idx = 0 if idx == 1 else 1 node.replace_all_uses_with(node, var_idx) logger.debug(f"removing {node.op} op: {node.name}")