model_pull.py

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)