Skip to content

Commit d44c3ca

Browse files
authored
Do not copy string when converting Term into EncodeInput (#60)
1 parent 3961417 commit d44c3ca

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

native/ex_tokenizers/src/tokenizer.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -413,10 +413,10 @@ pub fn tokenizer_disable_padding(tokenizer: ExTokenizersTokenizer) -> ExTokenize
413413
// / Inference
414414
// /////////////////////////////////////////////////////////////////////////////
415415

416-
fn term_to_encode_input<'a>(term: &'a Term) -> Result<EncodeInput<'a>, ExTokenizersError> {
417-
if let Ok(seq) = term.decode::<String>() {
416+
fn term_to_encode_input<'a, 'b>(term: &'a Term<'b>) -> Result<EncodeInput<'b>, ExTokenizersError> {
417+
if let Ok(seq) = term.decode::<&'b str>() {
418418
Ok(EncodeInput::Single(seq.into()))
419-
} else if let Ok((seq1, seq2)) = term.decode::<(String, String)>() {
419+
} else if let Ok((seq1, seq2)) = term.decode::<(&'b str, &'b str)>() {
420420
Ok(EncodeInput::Dual(seq1.into(), seq2.into()))
421421
} else {
422422
Err(ExTokenizersError::Other(String::from(

0 commit comments

Comments
 (0)