Unverified Commit ecd53971 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Fix RemBERT (#13375)

parent 33b7c9a8
...@@ -54,6 +54,7 @@ logger = logging.get_logger(__name__) ...@@ -54,6 +54,7 @@ logger = logging.get_logger(__name__)
_CONFIG_FOR_DOC = "RemBertConfig" _CONFIG_FOR_DOC = "RemBertConfig"
_TOKENIZER_FOR_DOC = "RemBertTokenizer" _TOKENIZER_FOR_DOC = "RemBertTokenizer"
_CHECKPOINT_FOR_DOC = "google/rembert"
REMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = [ REMBERT_PRETRAINED_MODEL_ARCHIVE_LIST = [
"google/rembert", "google/rembert",
......
...@@ -100,7 +100,7 @@ class RemBertTokenizerFast(PreTrainedTokenizerFast): ...@@ -100,7 +100,7 @@ class RemBertTokenizerFast(PreTrainedTokenizerFast):
def __init__( def __init__(
self, self,
vocab_file, vocab_file=None,
tokenizer_file=None, tokenizer_file=None,
do_lower_case=True, do_lower_case=True,
remove_space=True, remove_space=True,
......
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