From c432471546f931637abd0fe7fccc295ab0d78c03 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 4 Oct 2023 16:25:46 +0200
Subject: [PATCH] Fmt

---
 router/src/validation.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/router/src/validation.rs b/router/src/validation.rs
index 96d8b6d2..9adedc5b 100644
--- a/router/src/validation.rs
+++ b/router/src/validation.rs
@@ -84,9 +84,9 @@ impl Validation {
             let (inputs, input_length) = response_receiver.await.unwrap()?;
 
             // Get total tokens
-            let max_new_tokens: u32 = if let Some(max_new_tokens) = max_new_tokens{
+            let max_new_tokens: u32 = if let Some(max_new_tokens) = max_new_tokens {
                 max_new_tokens
-            }else{
+            } else {
                 self.max_total_tokens.saturating_sub(input_length) as u32
             };
             let total_tokens = input_length + max_new_tokens as usize;
@@ -117,9 +117,9 @@ impl Validation {
             // However, the inputs will be truncated by the python servers
             // We make sure that truncate + max_new_tokens <= self.max_total_tokens
             let input_length = truncate.unwrap_or(self.max_input_length);
-            let max_new_tokens: u32 = if let Some(max_new_tokens) = max_new_tokens{
+            let max_new_tokens: u32 = if let Some(max_new_tokens) = max_new_tokens {
                 max_new_tokens
-            }else{
+            } else {
                 self.max_total_tokens.saturating_sub(input_length) as u32
             };