move model for export preparation to be able reuse in other exporters (…

…#1698) * move model for export preparation to be able reuse in other exporters * rename to utils.py * Fix register decorator for task with past * exporter_type -> exporter * fix style * add backward compatibility and warnings
huggingface · Feb 27, 2024 · 80e89f1 · 80e89f1
1 parent e0cbf7d
commit 80e89f1
Show file tree

Hide file tree

Showing 4 changed files with 638 additions and 486 deletions.
diff --git a/examples/onnxruntime/training/image-classification/run_image_classification.py b/examples/onnxruntime/training/image-classification/run_image_classification.py
@@ -51,7 +51,6 @@
 from optimum.onnxruntime import ORTTrainer, ORTTrainingArguments
 
 
-
 """ Fine-tuning a 🤗 Transformers model for image classification"""
 
 logger = logging.getLogger(__name__)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -51,7 +51,6 @@
		from optimum.onnxruntime import ORTTrainer, ORTTrainingArguments



		""" Fine-tuning a 🤗 Transformers model for image classification"""

		logger = logging.getLogger(__name__)
Expand Down