From 84bf7eefbf9c5980372cc6c0c5cd39e10aea04a8 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Tue, 14 Jan 2025 12:13:40 +0100 Subject: [PATCH] style --- optimum/onnxruntime/trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/optimum/onnxruntime/trainer.py b/optimum/onnxruntime/trainer.py index 7304374a74..47a98e19c8 100644 --- a/optimum/onnxruntime/trainer.py +++ b/optimum/onnxruntime/trainer.py @@ -1097,7 +1097,6 @@ def _save(self, output_dir: Optional[str] = None, state_dict=None): os.makedirs(output_dir, exist_ok=True) logger.info(f"Saving model checkpoint to {output_dir}") - supported_classes = (PreTrainedModel,) # Save a trained model and configuration using `save_pretrained()`. # They can then be reloaded using `from_pretrained()`