File size: 236 Bytes
058d629
 
 
 
 
01554a7
1
2
3
4
5
6
from transformers import AutoModelForCausalLM, AutoTokenizer


model_checkpoint = "as-cle-bert/tiny-fungal-llama"
model = AutoModelForCausalLM.from_pretrained(model_checkpoint)
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)