fix style

huggingface · Mar 22, 2024 · 22bf6cc · 22bf6cc
1 parent 760c947
commit 22bf6cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/onnxruntime/modeling_ort.py b/optimum/onnxruntime/modeling_ort.py
@@ -43,6 +43,7 @@
 from transformers.modeling_outputs import (
     BaseModelOutput,
     CausalLMOutput,
+    GenerationMixin,
     ImageClassifierOutput,
     MaskedLMOutput,
     ModelOutput,
@@ -52,7 +53,6 @@
     SequenceClassifierOutput,
     TokenClassifierOutput,
     XVectorOutput,
-    GenerationMixin,
 )
 
 import onnxruntime as ort