From 68334a5fbff17525d187bb65141cf9a88b458d81 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Fri, 6 Dec 2024 05:38:05 +0100
Subject: [PATCH] Fixing the tests.

---
 backends/v2/src/queue.rs | 1 +
 backends/v3/src/queue.rs | 1 +
 2 files changed, 2 insertions(+)

diff --git a/backends/v2/src/queue.rs b/backends/v2/src/queue.rs
index bf52900f..61a3eebc 100644
--- a/backends/v2/src/queue.rs
+++ b/backends/v2/src/queue.rs
@@ -436,6 +436,7 @@ mod tests {
                 stopping_parameters: ValidStoppingParameters {
                     ignore_eos_token: false,
                     max_new_tokens: 1,
+                    max_total_new_tokens: 1024,
                     stop_sequences: vec![],
                 },
                 top_n_tokens: 0,
diff --git a/backends/v3/src/queue.rs b/backends/v3/src/queue.rs
index 6662b8de..dd27806f 100644
--- a/backends/v3/src/queue.rs
+++ b/backends/v3/src/queue.rs
@@ -573,6 +573,7 @@ mod tests {
                 stopping_parameters: ValidStoppingParameters {
                     ignore_eos_token: false,
                     max_new_tokens: 1,
+                    max_total_new_tokens: 1024,
                     stop_sequences: vec![],
                 },
                 top_n_tokens: 0,