Spaces:
Running
Running
File size: 279 Bytes
cbe8851 bbe4a78 85e6183 |
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, trust_remote_code=True)
# model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, trust_remote_code=True) |