From 7a6a7ed27bdb1ea8ffb0b10b7bd7de06c82c2460 Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Wed, 8 Mar 2023 17:57:54 +0100 Subject: [PATCH] black --- server/text_generation_server/models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/text_generation_server/models/model.py b/server/text_generation_server/models/model.py index d0645959..532e4060 100644 --- a/server/text_generation_server/models/model.py +++ b/server/text_generation_server/models/model.py @@ -35,4 +35,4 @@ class Model(ABC): [self.special_decode_token_id, token_id], skip_special_tokens=False ) # slice to remove special decode token - return result[self.special_decode_token_length:] + return result[self.special_decode_token_length :]