ankahi / test_tokenizers_lib.py
bhriguverma's picture
Add files using upload-large-folder tool
dc7e099 verified
Raw
History Blame Contribute Delete
392 Bytes
from tokenizers import Tokenizer
import os
MODEL_PATH = "/workspace/ankahi/punjabi_gemma/ankahi"
TOKENIZER_JSON = os.path.join(MODEL_PATH, "tokenizer.json")
print(f"Loading tokenizer from {TOKENIZER_JSON}")
try:
tokenizer = Tokenizer.from_file(TOKENIZER_JSON)
print("Success!")
print(f"Vocab size: {tokenizer.get_vocab_size()}")
except Exception as e:
print(f"Failed: {e}")