14e25f2 55f2d93 14e25f2
1
2
3
4
5
6
7
from transformers import AutoModelForCausalLM, AutoTokenizer def load_model(): model_name = "Erfan11/Neuracraft" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForCausalLM.from_pretrained(model_name) return model, tokenizer