Added a way to take the default special token list
This commit is contained in:
parent
8f5e2f2f0d
commit
3b274ad807
4
Project_Model/Libs/BPE/Utils/default_special_tokens.py
Normal file
4
Project_Model/Libs/BPE/Utils/default_special_tokens.py
Normal file
@ -0,0 +1,4 @@
|
||||
from ..Enums import SpecialToken
|
||||
|
||||
def default_special_tokens() -> list[str]:
|
||||
return [token.value for token in SpecialToken]
|
||||
Loading…
x
Reference in New Issue
Block a user