File size: 939 Bytes
8265028
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# ---------- check_tokenizer_size.py ----------
# (์ฃผ์‹ํšŒ์‚ฌ์— ์ผ€์ด๋”” | MKD)
# Display total vocab tokens and a few sample tokens

from tokenizers import Tokenizer
import os

BASE_DIR = "/home/mkd-s003/AI/finetune"
TOKENIZER_PATH = os.path.join(BASE_DIR, "prod_tokenizer_v2.json")

print("๐Ÿ” Checking tokenizer information...\n")

# Load tokenizer
tok = Tokenizer.from_file(TOKENIZER_PATH)
vocab = tok.get_vocab()
vocab_size = len(vocab)

print(f"โœ… Tokenizer loaded: {TOKENIZER_PATH}")
print(f"๐Ÿ“˜ Total vocabulary size: {vocab_size:,} tokens\n")

# Show first few sample tokens
sample_tokens = list(vocab.keys())[:20]
print("๐Ÿ”น Sample tokens:")
for t in sample_tokens:
    print(f"   {t}")

# Check if special tokens exist
specials = ["<|endoftext|>", "<|pad|>", "<|unk|>", "<|im_start|>", "<|im_end|>"]
print("\n๐Ÿ”น Special tokens present:")
for s in specials:
    print(f"   {s}: {'โœ… Yes' if s in vocab else 'โŒ No'}")