From 9099f837cfae1a87f0c5efd04f13201da05c59ba Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Thu, 16 Jan 2025 11:33:55 +0000 Subject: [PATCH] fix type hint --- src/transformers/generation/streamers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/generation/streamers.py b/src/transformers/generation/streamers.py index c78e259db38b..e63f29a7442a 100644 --- a/src/transformers/generation/streamers.py +++ b/src/transformers/generation/streamers.py @@ -282,7 +282,7 @@ class AsyncTextIteratorStreamer(TextStreamer): """ def __init__( - self, tokenizer: "AutoTokenizer", skip_prompt: bool = False, timeout: float | None = None, **decode_kwargs + self, tokenizer: "AutoTokenizer", skip_prompt: bool = False, timeout: Optional[float] = None, **decode_kwargs ): super().__init__(tokenizer, skip_prompt, **decode_kwargs) self.text_queue = asyncio.Queue()