Views
No views yet
meta-llama/Meta-Llama-3-8B following the BK-MOO-Pretrain recipe (https://github.com/longhp1618/BK-MOO-Pretrain).cagradfinewebmeta-llama/Meta-Llama-3-8B*_limas1 stages)1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3repo = "Yiran0924/bk-moo-llama3-8b-cagrad-fineweb"
4tok = AutoTokenizer.from_pretrained(repo)
5model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype="bfloat16", device_map="auto")