Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
84acf0c7
Unverified
Commit
84acf0c7
authored
Feb 08, 2021
by
demSd
Committed by
GitHub
Feb 08, 2021
Browse files
remove token_type_ids from TokenizerBertGeneration output (#10070)
parent
e4bf9910
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
0 deletions
+1
-0
src/transformers/models/bert_generation/tokenization_bert_generation.py
...rs/models/bert_generation/tokenization_bert_generation.py
+1
-0
No files found.
src/transformers/models/bert_generation/tokenization_bert_generation.py
View file @
84acf0c7
...
...
@@ -58,6 +58,7 @@ class BertGenerationTokenizer(PreTrainedTokenizer):
pretrained_vocab_files_map
=
{
"vocab_file"
:
{
"bert_for_seq_generation"
:
tokenizer_url
}}
max_model_input_sizes
=
{
"bert_for_seq_generation"
:
512
}
prefix_tokens
:
List
[
int
]
=
[]
model_input_names
=
[
"input_ids"
,
"attention_mask"
]
def
__init__
(
self
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment