File size: 229 Bytes
cbe8851
 
 
 
 
 
 
1
2
3
4
5
6
7
from transformers import AutoModelForCausalLM
from transformers import AutoTokenizer

BASE_MODEL = "pfnet/plamo-2-1b"

tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
model = AutoModelForCausalLM.from_pretrained(BASE_MODEL)