Remove unused imports.

mfuntowicz · mfuntowicz · commit ea01db414db2 · 2021-07-01T15:26:01.000+02:00
diff --git a/src/transformers/onnx/__main__.py b/src/transformers/onnx/__main__.py
@@ -38,14 +38,14 @@
 FRAMEWORK_CHOICES = {FRAMEWORK_NAME_PT, FRAMEWORK_NAME_PT}
 
 if is_torch_available():
-    from transformers import AutoModel, AutoModelForTokenClassification, PreTrainedModel
+    from transformers import AutoModel, PreTrainedModel
 
     FEATURES_TO_AUTOMODELS = {
         "default": AutoModel,
     }
 
 if is_tf_available():
-    from transformers import TFAutoModel, TFAutoModelForTokenClassification, TFPreTrainedModel
+    from transformers import TFAutoModel, TFPreTrainedModel
 
     FEATURES_TO_TF_AUTOMODELS = {
         "default": TFAutoModel,
