nit

huggingface · Jul 1, 2024 · d82efd3 · d82efd3
1 parent 4309f35
commit d82efd3
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py
@@ -316,7 +316,6 @@ class MistralOnnxConfig(TextDecoderWithPositionIdsOnnxConfig):
     def patch_model_for_export(
         self, model: Union["PreTrainedModel", "TFPreTrainedModel"], model_kwargs: Optional[Dict[str, Any]] = None
     ) -> "ModelPatcher":
-        print("IN patch_model_for_export")
         return MistralModelPatcher(self, model, model_kwargs=model_kwargs)
 
 

diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
@@ -1077,7 +1077,6 @@ def __enter__(self):
                 "_prepare_4d_causal_attention_mask_for_sdpa", _prepare_4d_causal_attention_mask_for_sdpa_patched
             )
 
-        print("self._model in enter", self._model)
         if _transformers_version >= version.parse("4.42"):
             if hasattr(self._model, "model"):
                 self._model.model._update_causal_mask = types.MethodType(