File tree 1 file changed +4
-4
lines changed
src/transformers/models/prism
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -169,11 +169,9 @@ def __init__(
169
169
fairseq_language_code = FAIRSEQ_LANGUAGE_CODES [language_codes ]
170
170
self .lang_code_to_token = {lang_code : f"<{ lang_code } >" for lang_code in fairseq_language_code }
171
171
172
+ language_tokens = [self .get_lang_token (lang_code ) for lang_code in fairseq_language_code ]
172
173
additional_special_tokens = kwargs .pop ("additional_special_tokens" , [])
173
- for lang_code in fairseq_language_code :
174
- token = self .get_lang_token (lang_code )
175
- if token not in additional_special_tokens and lang_code not in str (token ) not in self .added_tokens_encoder :
176
- additional_special_tokens .append (token )
174
+ self .additional_special_tokens = language_tokens + additional_special_tokens
177
175
178
176
self .vocab_file = vocab_file
179
177
self .encoder = load_json (vocab_file )
@@ -215,6 +213,8 @@ def __init__(
215
213
num_madeup_words = num_madeup_words ,
216
214
** kwargs ,
217
215
)
216
+
217
+ self .special_tokens_map ['additional_special_tokens' ] = self .additional_special_tokens
218
218
self .set_src_lang_special_tokens (self ._src_lang )
219
219
220
220
@property
You can’t perform that action at this time.
0 commit comments