Merge pull request #1152 from epwalsh/fix-special-tokens

fix adding special tokens
This commit is contained in:
Thomas Wolf authored and GitHub committed 2019-08-30 23:21:59 +02:00
commit f3d18c71ec
1 file changed
+1 -1
+1 -1
View File
@@ -784,7 +784,7 @@ class PreTrainedTokenizer(object):
all_toks = []
set_attr = self.special_tokens_map
for attr_value in set_attr.values():
all_toks = all_toks + (attr_value if isinstance(attr_value, (list, tuple)) else [attr_value])
all_toks = all_toks + (list(attr_value) if isinstance(attr_value, (list, tuple)) else [attr_value])
all_toks = list(set(all_toks))
return all_toks