Fix dummy objects for quantization (#14478)

* Fix dummy objects for quantization

* Add more models
This commit is contained in:
Sylvain Gugger
2021-11-21 17:39:20 -05:00
committed by GitHub
parent c9d2cf855a
commit 1a92bc5788
5 changed files with 89 additions and 1 deletions

View File

@@ -23,7 +23,7 @@ import re
PATH_TO_TRANSFORMERS = "src/transformers"
# Matches is_xxx_available()
_re_backend = re.compile(r"is\_([a-z]*)_available()")
_re_backend = re.compile(r"is\_([a-z_]*)_available()")
# Matches from xxx import bla
_re_single_line_import = re.compile(r"\s+from\s+\S*\s+import\s+([^\(\s].*)\n")
_re_test_backend = re.compile(r"^\s+if\s+is\_[a-z]*\_available\(\)")
@@ -131,6 +131,7 @@ def create_dummy_object(name, backend_name):
"ForConditionalGeneration",
"ForMaskedLM",
"ForMultipleChoice",
"ForNextSentencePrediction",
"ForObjectDetection",
"ForQuestionAnswering",
"ForSegmentation",