Skip to content

fix: Support Kimi Tokenizer using allow_special_tokens [TL-76] - #420

Open
ajcasagrande wants to merge 1 commit into
mainfrom
ajc/fix-kimi-tokenizer
Open

fix: Support Kimi Tokenizer using allow_special_tokens [TL-76]#420
ajcasagrande wants to merge 1 commit into
mainfrom
ajc/fix-kimi-tokenizer

Conversation

@ajcasagrande

Copy link
Copy Markdown
Contributor

Fixes to tokenizer functionality:

  • Added a new supports_kwarg function in genai-perf/genai_perf/utils.py to determine if a method of an object accepts a specific keyword argument.

  • Updated the set_tokenizer method in genai-perf/genai_perf/tokenizer.py to use the supports_kwarg utility. If the tokenizer's encode method supports the allow_special_tokens argument, the method dynamically adjusts the call arguments (_call_args, _encode_args, _decode_args) to align with the tokenizer's behavior.

@ajcasagrande ajcasagrande changed the title fix: Support Kimi Tokenizer using allow_special_tokens fix: Support Kimi Tokenizer using allow_special_tokens [TL-76] Jul 17, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant