pcuenq HF Staff commited on
Commit
dca5adf
·
verified ·
1 Parent(s): 9659d5d

Compatibility with transformers v5

Browse files
Files changed (1) hide show
  1. tokenization_kimi.py +6 -1
tokenization_kimi.py CHANGED
@@ -8,7 +8,12 @@ from typing import Any, Dict, Iterator, List, Optional, Tuple, Union, cast
8
  import tiktoken
9
  from tiktoken.load import load_tiktoken_bpe
10
  from tokenizers import AddedToken
11
- from transformers.models.gpt2.tokenization_gpt2 import bytes_to_unicode
 
 
 
 
 
12
  from transformers.tokenization_utils import PreTrainedTokenizer
13
 
14
  from .tool_declaration_ts import encode_tools_to_typescript_style
 
8
  import tiktoken
9
  from tiktoken.load import load_tiktoken_bpe
10
  from tokenizers import AddedToken
11
+
12
+ try:
13
+ from transformers.models.gpt2.tokenization_gpt2 import bytes_to_unicode
14
+ except:
15
+ from transformers.convert_slow_tokenizer import bytes_to_unicode
16
+
17
  from transformers.tokenization_utils import PreTrainedTokenizer
18
 
19
  from .tool_declaration_ts import encode_tools_to_typescript_style