tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix typing list

Browse Source
This commit is contained in:
NanoCode012
2023-06-29 14:29:55 +09:00
committed by GitHub
parent f150c027e3
commit 77bdb7d144
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/prompt_tokenizers.py
View File

@@ -440,7 +440,7 @@ def parse_tokenized_to_result(
result: Dict[str, List[int]], result: Dict[str, List[int]],
current_len: int, current_len: int,
res: Dict[str, List[int]], res: Dict[str, List[int]],
labels: list[int], labels: List[int],
pad_token_id: Union[int, None] = None, pad_token_id: Union[int, None] = None,
) -> Tuple[Dict[str, List[int]], int]: ) -> Tuple[Dict[str, List[int]], int]:
""" """