SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[generation] fix type hint (#35725)

Browse Source 
fix type hint
This commit is contained in:
Joao Gante
2025-01-16 15:09:59 +00:00
committed by GitHub
parent 8b78d9d6e7
commit 3292e96a4f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/generation/streamers.py
View File

@@ -284,7 +284,7 @@ class AsyncTextIteratorStreamer(TextStreamer):
""" """
def __init__( 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) super().__init__(tokenizer, skip_prompt, **decode_kwargs)
self.text_queue = asyncio.Queue() self.text_queue = asyncio.Queue()