diff --git a/modelopt/onnx/quantization/quantize.py b/modelopt/onnx/quantization/quantize.py index c00e3ca37d..6349c128e9 100755 --- a/modelopt/onnx/quantization/quantize.py +++ b/modelopt/onnx/quantization/quantize.py @@ -528,10 +528,6 @@ def quantize( ) if direct_io_types: onnx_model = remove_graph_input_q(onnx_model) - # Sort nodes topologically - graph = gs.import_onnx(onnx_model) - graph.toposort().cleanup() - onnx_model = gs.export_onnx(graph) else: # Remove redundant cast nodes in the quantized model # Note. This is called within the qdq_to_dq function as well