From 77bdb7d1444cd0fbd822a1a68fc2db6abbb78814 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Thu, 29 Jun 2023 14:29:55 +0900 Subject: [PATCH] Fix typing list --- src/axolotl/prompt_tokenizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/prompt_tokenizers.py b/src/axolotl/prompt_tokenizers.py index cf80539eb..8216d73dd 100644 --- a/src/axolotl/prompt_tokenizers.py +++ b/src/axolotl/prompt_tokenizers.py @@ -440,7 +440,7 @@ def parse_tokenized_to_result( result: Dict[str, List[int]], current_len: int, res: Dict[str, List[int]], - labels: list[int], + labels: List[int], pad_token_id: Union[int, None] = None, ) -> Tuple[Dict[str, List[int]], int]: """