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

Lianmin Zheng's avatar
Lianmin Zheng committed
4
5
#name = "meta-llama/Llama-2-7b-chat-hf"
name = "meta-llama/Meta-Llama-3-8B-Instruct"
Lianmin Zheng's avatar
Lianmin Zheng committed
6
7
8

t = transformers.AutoTokenizer.from_pretrained(name)
code.interact(local=locals())