From be7835475b10bd2f2c744ee59ee4230129a83eb0 Mon Sep 17 00:00:00 2001 From: drbh Date: Wed, 14 Feb 2024 00:47:28 +0000 Subject: [PATCH] fix: add missing grammar type typo --- benchmark/src/lib.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/benchmark/src/lib.rs b/benchmark/src/lib.rs index 595e0cb9..638c6514 100644 --- a/benchmark/src/lib.rs +++ b/benchmark/src/lib.rs @@ -8,7 +8,7 @@ use crate::app::App; use crate::event::Event; use crossterm::ExecutableCommand; use std::io; -use text_generation_client::{NextTokenChooserParameters, ShardedClient}; +use text_generation_client::{GrammarType, NextTokenChooserParameters, ShardedClient}; use tokenizers::Tokenizer; use tokio::sync::{broadcast, mpsc}; use tui::backend::CrosstermBackend; @@ -46,6 +46,7 @@ pub async fn run( frequency_penalty: frequency_penalty.unwrap_or(0.0), watermark, grammar: String::new(), + grammar_type: GrammarType::None as i32, }; // Initialize terminal properties