Skip to content

GitLab

Unverified Commit 5a386fb0 authored by Pavel Belevich's avatar Pavel Belevich Committed by GitHub
Browse files

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

parent a7aca42f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
src/transformers/utils/fx.py
View file @ 5a386fb0
...@@ -115,7 +115,9 @@ _SPECIAL_SUPPORTED_MODELS = [ ...@@ -115,7 +115,9 @@ _SPECIAL_SUPPORTED_MODELS = [
# TODO: add support for them as it should be quite easy to do so (small blocking issues). # TODO: add support for them as it should be quite easy to do so (small blocking issues).
# XLNetForQuestionAnswering, # 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): class HFProxy(Proxy):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment