Minor fix

Signed-off-by: Riyad Islam <[email protected]>

Author: i-riyad

modelopt/onnx/utils.py

@@ -779,8 +779,8 @@ def remove_node_training_mode(onnx_model: onnx.ModelProto, node_op_type: str) ->
             removed_output_names.update(node.output[1:])
             node.output[:] = node.output[:1]
 
-    # Clean up corresponding value_info entries
     if removed_output_names:
+        # Clean up corresponding value_info entries
         keep = [vi for vi in onnx_model.graph.value_info if vi.name not in removed_output_names]
         del onnx_model.graph.value_info[:]
         onnx_model.graph.value_info.extend(keep)