Skip to content

Commit 96d6684

Browse files
committed
Add return type annotation to load_tokenizer
1 parent 41c1d88 commit 96d6684

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlx_lm/tokenizer_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -428,7 +428,7 @@ def load_tokenizer(
428428
tokenizer_config_extra: Optional[Dict[str, Any]]=None,
429429
return_tokenizer=True,
430430
eos_token_ids=None
431-
):
431+
) -> TokenizerWrapper:
432432
"""Load a huggingface tokenizer and try to infer the type of streaming
433433
detokenizer to use.
434434

0 commit comments

Comments
 (0)