Update optimum/onnxruntime/trainer_seq2seq.py

Co-authored-by: fxmarty <[email protected]>
huggingface · Sep 18, 2023 · da18b6e · da18b6e
1 parent fec39e3
commit da18b6e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/optimum/onnxruntime/trainer_seq2seq.py b/optimum/onnxruntime/trainer_seq2seq.py
@@ -51,6 +51,10 @@
 
 if is_accelerate_available():
     from accelerate import __version__ as accelerate_version
+else:
+    raise ImportError(
+        "The package `accelerate` is required to use the ORTTrainer. Please install it following https://huggingface.co/docs/accelerate/basic_tutorials/install."
+    )
 
 logger = logging.get_logger(__name__)