Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load any checkpoint
4model = AutoModelForCausalLM.from_pretrained("./checkpoint_epoch0_batch2")
5tokenizer = AutoTokenizer.from_pretrained("./checkpoint_epoch0_batch2")