Use forward references for type hints, avoid NameError (#6780)

This commit is contained in:
Vladimir Blagojevic 2024-01-18 18:46:59 +01:00 committed by GitHub
parent d4f6531c52
commit f47439c2a2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -272,7 +272,7 @@ class HuggingFaceLocalChatGenerator:
self,
text: str,
index: int,
tokenizer: Union[PreTrainedTokenizer, PreTrainedTokenizerFast],
tokenizer: Union["PreTrainedTokenizer", "PreTrainedTokenizerFast"],
prompt: str,
generation_kwargs: Dict[str, Any],
) -> ChatMessage: