load_tokenizer.py 365 Bytes
Newer Older
1
import argparse
Lianmin Zheng's avatar
Lianmin Zheng committed
2
3
import code

4
from sglang.srt.utils.hf_transformers_utils import get_tokenizer
Lianmin Zheng's avatar
Lianmin Zheng committed
5

6
7
if __name__ == "__main__":
    parser = argparse.ArgumentParser()
zhyncs's avatar
zhyncs committed
8
9
10
    parser.add_argument(
        "--name", type=str, default="meta-llama/Meta-Llama-3-8B-Instruct"
    )
11
12
13
    args = parser.parse_args()

    t = get_tokenizer(args.name)
zhyncs's avatar
zhyncs committed
14
    code.interact(local=locals())