Make transformers.utils.fx. _SUPPORTED_MODELS unique (#16015)

5a386fb0 · Pavel Belevich · GitHub · a7aca42f · 5a386fb0
Unverified Commit 5a386fb0 authored Mar 15, 2022 by Pavel Belevich Committed by GitHub Mar 15, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

src/transformers/utils/fx.py src/transformers/utils/fx.py +3 -1

No files found.
--- a/src/transformers/utils/fx.py
+++ b/src/transformers/utils/fx.py
@@ -115,7 +115,9 @@ _SPECIAL_SUPPORTED_MODELS = [
    # TODO: add support for them as it should be quite easy to do so (small blocking issues).
    # XLNetForQuestionAnswering,
 ]
-_SUPPORTED_MODELS = tuple(_REGULAR_SUPPORTED_MODELS + _SPECIAL_SUPPORTED_MODELS)
+_SUPPORTED_MODELS = tuple(
+    sorted(list(set(_REGULAR_SUPPORTED_MODELS + _SPECIAL_SUPPORTED_MODELS)), key=lambda c: c.__name__)
+)


 class HFProxy(Proxy):