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

Python - Add WordPieceTrainer

Browse Source
This commit is contained in:
Anthony MOI
2020-01-03 16:37:36 -05:00
parent e64b54b29e
commit c51e340492
2 changed files with 51 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
bindings/python/src/lib.rs
View File

@@ -18,6 +18,7 @@ use pyo3::wrap_pymodule;
fn trainers(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_class::<trainers::Trainer>()?;
m.add_class::<trainers::BpeTrainer>()?;
m.add_class::<trainers::WordPieceTrainer>()?;
Ok(())
}