[Seq2Seq Templates] Fix check_repo.py templates file (#9277)
* add enc dec pt model to check repo * fix indent
This commit is contained in:
committed by
GitHub
parent
4bafc43b0e
commit
d5db6c37d4
@@ -299,3 +299,23 @@ from ..{{cookiecutter.lowercase_modelname}}.modeling_tf_{{cookiecutter.lowercase
|
|||||||
({{cookiecutter.camelcase_modelname}}Config, TF{{cookiecutter.camelcase_modelname}}ForConditionalGeneration),
|
({{cookiecutter.camelcase_modelname}}Config, TF{{cookiecutter.camelcase_modelname}}ForConditionalGeneration),
|
||||||
{% endif -%}
|
{% endif -%}
|
||||||
# End.
|
# End.
|
||||||
|
|
||||||
|
# To replace in: "utils/check_repo.py" if generating PyTorch
|
||||||
|
|
||||||
|
# Below: "models to ignore for model xxx mapping"
|
||||||
|
# Replace with:
|
||||||
|
{% if cookiecutter.is_encoder_decoder_model == "False" -%}
|
||||||
|
{% else -%}
|
||||||
|
"{{cookiecutter.camelcase_modelname}}Encoder",
|
||||||
|
"{{cookiecutter.camelcase_modelname}}Decoder",
|
||||||
|
{% endif -%}
|
||||||
|
# End.
|
||||||
|
|
||||||
|
# Below: "models to ignore for not tested"
|
||||||
|
# Replace with:
|
||||||
|
{% if cookiecutter.is_encoder_decoder_model == "False" -%}
|
||||||
|
{% else -%}
|
||||||
|
"{{cookiecutter.camelcase_modelname}}Encoder", # Building part of bigger (tested) model.
|
||||||
|
"{{cookiecutter.camelcase_modelname}}Decoder", # Building part of bigger (tested) model.
|
||||||
|
{% endif -%}
|
||||||
|
# End.
|
||||||
|
|||||||
@@ -29,6 +29,7 @@ PATH_TO_DOC = "docs/source"
|
|||||||
# Update this list for models that are not tested with a comment explaining the reason it should not be.
|
# Update this list for models that are not tested with a comment explaining the reason it should not be.
|
||||||
# Being in this list is an exception and should **not** be the rule.
|
# Being in this list is an exception and should **not** be the rule.
|
||||||
IGNORE_NON_TESTED = [
|
IGNORE_NON_TESTED = [
|
||||||
|
# models to ignore for not tested
|
||||||
"BartDecoder", # Building part of bigger (tested) model.
|
"BartDecoder", # Building part of bigger (tested) model.
|
||||||
"BartEncoder", # Building part of bigger (tested) model.
|
"BartEncoder", # Building part of bigger (tested) model.
|
||||||
"BertLMHeadModel", # Needs to be setup as decoder.
|
"BertLMHeadModel", # Needs to be setup as decoder.
|
||||||
@@ -62,6 +63,7 @@ TEST_FILES_WITH_NO_COMMON_TESTS = [
|
|||||||
# Update this list for models that are not in any of the auto MODEL_XXX_MAPPING. Being in this list is an exception and
|
# Update this list for models that are not in any of the auto MODEL_XXX_MAPPING. Being in this list is an exception and
|
||||||
# should **not** be the rule.
|
# should **not** be the rule.
|
||||||
IGNORE_NON_AUTO_CONFIGURED = [
|
IGNORE_NON_AUTO_CONFIGURED = [
|
||||||
|
# models to ignore for model xxx mapping
|
||||||
"BartDecoder",
|
"BartDecoder",
|
||||||
"BartEncoder",
|
"BartEncoder",
|
||||||
"DPRContextEncoder",
|
"DPRContextEncoder",
|
||||||
|
|||||||
Reference in New Issue
Block a user