From 8b13c47d6348e505039424d40715a4e94d4acd1e Mon Sep 17 00:00:00 2001
From: drbh <david.richard.holtz@gmail.com>
Date: Wed, 14 Feb 2024 00:31:50 +0000
Subject: [PATCH] fix: adjust type typo

---
 server/text_generation_server/utils/tokens.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/text_generation_server/utils/tokens.py b/server/text_generation_server/utils/tokens.py
index 8f2ae4f6..896d4d40 100644
--- a/server/text_generation_server/utils/tokens.py
+++ b/server/text_generation_server/utils/tokens.py
@@ -232,7 +232,7 @@ class HeterogeneousNextTokenChooser:
         seeds: List[int],
         tokenizer: PreTrainedTokenizerBase,
         grammars: List[str],
-        grammar_types: List[GrammarType],
+        grammar_types: List[int],
         fsm_grammar_states=List[int],
     ):
         warpers = []