This commit is contained in:
Arthur Zucker
2023-09-05 16:43:06 +00:00
parent 26fdfc2bc3
commit f435af8b71

View File

@ -669,10 +669,13 @@ impl PyTokenizer {
#[pyo3(signature = ())]
#[pyo3(text_signature = "(self)")]
fn get_added_tokens_decoder(&self) -> HashMap<u32, PyAddedToken> {
self.tokenizer.get_added_tokens_decoder().into_iter().map(|(key, value)| (key, value.into())).collect()
self.tokenizer
.get_added_tokens_decoder()
.into_iter()
.map(|(key, value)| (key, value.into()))
.collect()
}
/// Get the size of the underlying vocabulary
///
/// Args: