Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3INTERMEDIATE_STEP = "0000120000"
4model = AutoModelForCausalLM.from_pretrained('trillionlabs/Tri-7B-Intermediate-Checkpoints', revision=INTERMEDIATE_STEP, trust_remote_code=True)
5tokenizer = AutoTokenizer.from_pretrained('trillionlabs/Tri-7B-Intermediate-Checkpoints', revision=INTERMEDIATE_STEP, trust_remote_code=True)
6
7...