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

Add bindings for Trainer in Python

Browse Source
This commit is contained in:
Anthony MOI
2019-12-03 15:54:15 -05:00
parent 310a2af76b
commit eaafb22511
5 changed files with 74 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,11 +3,20 @@ mod models;
mod pre_tokenizers;
mod token;
mod tokenizer;
mod trainers;
mod utils;
use pyo3::prelude::*;
use pyo3::wrap_pymodule;
/// Trainers Module
#[pymodule]
fn trainers(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_class::<trainers::Trainer>()?;
m.add_class::<trainers::BpeTrainer>()?;
Ok(())
}
/// Models Module
#[pymodule]
fn models(_py: Python, m: &PyModule) -> PyResult<()> {
@@ -39,5 +48,6 @@ fn tokenizers(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_wrapped(wrap_pymodule!(models))?;
m.add_wrapped(wrap_pymodule!(pre_tokenizers))?;
m.add_wrapped(wrap_pymodule!(decoders))?;
m.add_wrapped(wrap_pymodule!(trainers))?;
Ok(())
}