Remove TFPreTrainedModel reference

mfuntowicz · mfuntowicz · commit 66d7160fc72f · 2021-07-01T16:03:22.000+02:00
diff --git a/src/transformers/onnx/__main__.py b/src/transformers/onnx/__main__.py
@@ -74,7 +74,7 @@ def get_model_from_features(features: str, model: str):
 
 
 def check_supported_model_or_raise(
-    model: Union[PreTrainedModel, TFPreTrainedModel], features: str = "default"
+    model: PreTrainedModel, features: str = "default"
 ) -> Tuple[str, Callable]:
     """
     Check whether or not the model has the requested features
