fix: adjust unwrap syntax in template

huggingface · Jun 28, 2024 · a00db1b · a00db1b
1 parent 8885688
commit a00db1b
Showing 1 changed file with 2 additions and 14 deletions.
diff --git a/router/src/infer/mod.rs b/router/src/infer/mod.rs
@@ -293,20 +293,8 @@ impl ChatTemplate {
 
         Self {
             template,
-            bos_token: match bos_token {
-                Some(token) => match token {
-                    TokenizerConfigToken::String(token) => Some(token),
-                    TokenizerConfigToken::Object { content } => Some(content),
-                },
-                None => None,
-            },
-            eos_token: match eos_token {
-                Some(token) => match token {
-                    TokenizerConfigToken::String(token) => Some(token),
-                    TokenizerConfigToken::Object { content } => Some(content),
-                },
-                None => None,
-            },
+            bos_token: bos_token.map(|token| token.as_str().to_string()),
+            eos_token: eos_token.map(|token| token.as_str().to_string()),
             use_default_tool_template,
         }
     }