Merge pull request #434 from IQTLabs/renovate/transformers-4.x-lockfile

Update dependency transformers to v4.40.2
IQTLabs · May 13, 2024 · cb8da72 · cb8da72
2 parents a9aa428 + 64db149
commit cb8da72
Show file tree

Hide file tree

Showing 3 changed files with 954 additions and 945 deletions.
diff --git a/daisybell/helpers/chat.py b/daisybell/helpers/chat.py
@@ -7,7 +7,7 @@ class StopOnTokens(StoppingCriteria):
     Stop the chat bot generation when one of the given tokens is generated.
     """
 
-    def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs) -> bool:
+    def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs) -> bool:  # type: ignore
         # These are the ids for stop tokens in StableLM and GPT-Neo
         # 50278 is <|ASSISTANT|>, 50279 is <|SYSTEM|>, 50277 is <|USER|>
         stop_ids = [