improves modelbuilder wrapper

.github/workflows/documentation.yml

@@ -109,9 +109,9 @@ jobs:
             grep ERROR doc.txt | grep -v 'l-plot-tiny-llm-export' | grep -v 'Unexpected section title or transition.'
             exit 1
           fi
-          if [[ $(grep WARNING doc.txt | grep -v 'l-plot-tiny-llm-export' | grep -v 'Inline emphasis start-string' | grep -v 'Definition list ends without a blank line' | grep -v 'Unexpected section title or transition' | grep -v 'Inline strong start-string' | grep -v 'MambaCache' ) ]]; then
+          if [[ $(grep WARNING doc.txt | grep -v 'l-plot-tiny-llm-export' | grep -v 'Inline emphasis start-string' | grep -v 'Definition list ends without a blank line' | grep -v 'Unexpected section title or transition' | grep -v 'Inline strong start-string' | grep -v 'MambaCache' | grep -v 'duplicate label patch-_patched_patch_transformers-common' ) ]]; then
             echo "Documentation produces warnings."
-            grep WARNING doc.txt | grep -v 'l-plot-tiny-llm-export' | grep -v 'Inline emphasis start-string' | grep -v 'Definition list ends without a blank line' | grep -v 'Unexpected section title or transition' | grep -v 'Inline strong start-string' | grep -v 'MambaCache'
+            grep WARNING doc.txt | grep -v 'l-plot-tiny-llm-export' | grep -v 'Inline emphasis start-string' | grep -v 'Definition list ends without a blank line' | grep -v 'Unexpected section title or transition' | grep -v 'Inline strong start-string' | grep -v 'MambaCache' | grep -v 'duplicate label patch-_patched_patch_transformers-common'
             exit 1
           fi
 

CHANGELOGS.rst

@@ -4,6 +4,7 @@ Change Logs
 0.9.1
 +++++
 
+* :pr:`409`: improves ModelBuilder wrapper
 * :pr:`408`: fix torch_deepcopy for empty DynamicCache and transformers==5.1.0, 5.2.0 (see https://github.com/huggingface/transformers/pull/43765/)
 
 0.9.0

onnx_diagnostic/helpers/model_builder_helper.py

@@ -310,7 +310,8 @@ def _post(onnx_model):
             if not hasattr(config, key):
                 setattr(config, key, getattr(text_config, key))
     elif config.architectures[0] == "GptOssForCausalLM":
-        delattr(config, "quantization_config")
+        if hasattr(config, "quantization_config"):
+            delattr(config, "quantization_config")
     elif (
         config.architectures[0] == "PhiMoEForCausalLM"
         and config.max_position_embeddings != config.original_max_position_embeddings