From 5b5cbb14d6bdc1d9ab566f3fdc71b57d537b06be Mon Sep 17 00:00:00 2001 From: drbh Date: Fri, 8 Mar 2024 04:18:31 +0000 Subject: [PATCH] fix: remove GrammarType import 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 1a65bdb80..07c8e0202 100644 --- a/server/text_generation_server/utils/tokens.py +++ b/server/text_generation_server/utils/tokens.py @@ -4,7 +4,7 @@ from typing import List, Optional, Tuple import math import torch from text_generation_server.pb import generate_pb2 -from text_generation_server.pb.generate_pb2 import FinishReason, GrammarType +from text_generation_server.pb.generate_pb2 import FinishReason from text_generation_server.utils.logits_process import ( FrequencyPenaltyLogitsProcessor, GrammarLogitProcessor,