huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 20:04:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fmt.

Browse Source
This commit is contained in:
Nicolas Patry 2023-08-14 18:28:57 +02:00
parent d73204d72c
commit 3a86e4f519
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
benchmark/src/generation.rs
View File

@ -217,7 +217,5 @@ fn create_sequence(sequence_length: u32, tokenizer: Tokenizer) -> String {
// Truncate to sequence_length // Truncate to sequence_length
encoding.truncate(sequence_length as usize, 0, TruncationDirection::Left); encoding.truncate(sequence_length as usize, 0, TruncationDirection::Left);
// Decode // Decode
tokenizer tokenizer.decode(encoding.get_ids(), false).unwrap()
.decode(encoding.get_ids(), false)
.unwrap()
} }