plamo-2-1b / model_pull.py
masanorihirano's picture
Update model_pull.py
85e6183 verified
raw
history blame
279 Bytes
from transformers import AutoModelForCausalLM
from transformers import AutoTokenizer
BASE_MODEL = "pfnet/plamo-2-1b"
tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, trust_remote_code=True)
# model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, trust_remote_code=True)