mii/tokenizers
1
0
Fork 0
mirror of https://github.com/mii443/tokenizers.git synced 2025-08-23 00:35:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow initial_alphabet on UnigramTrainer

Browse Source
This commit is contained in:
Anthony MOI
2020-10-22 14:32:40 -04:00
committed by Anthony MOI
parent f7c61c267a
commit 1a6f4b5204
3 changed files with 44 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
bindings/python/src/trainers.rs
View File

@ -193,6 +193,17 @@ impl PyUnigramTrainer {
"unk_token" => builder.unk_token(val.extract()?),
"max_piece_length" => builder.max_piece_length(val.extract()?),
"seed_size" => builder.seed_size(val.extract()?),
"initial_alphabet" => {
let alphabet: Vec<String> = val.extract()?;
builder.initial_alphabet(
alphabet
.into_iter()
.map(|s| s.chars().next())
.filter(|c| c.is_some())
.map(|c| c.unwrap())
.collect(),
)
}
"special_tokens" => builder.special_tokens(
val.cast_as::<PyList>()?
.into_iter()