Apply suggestions from code review

neuralmagic · Jan 24, 2024 · c45125a · c45125a
1 parent 807898f
commit c45125a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/sparsezoo/utils/onnx/external_data.py b/src/sparsezoo/utils/onnx/external_data.py
@@ -117,10 +117,10 @@ def save_onnx(
 
     if model.ByteSize() > DUMP_EXTERNAL_DATA_THRESHOLD:
         external_data_file = external_data_file or EXTERNAL_ONNX_DATA_NAME
-        _LOGGER.warning(
+        _LOGGER.debug(
             "The ONNX model is too large to be saved as a single protobuf. "
             "Saving with external data, with file chunks of maximum size "
-            f"{DUMP_EXTERNAL_DATA_THRESHOLD / 1e9} GB"
+            f"{max_external_file_size / 1e9} GB"
         )
         _check_for_old_external_data(
             model_path=model_path, external_data_file=external_data_file