load_model.py

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)