nursing-copilot-api / backend /model_loader.py
MarcoLeung052's picture
Update backend/model_loader.py
e0eadad verified
raw
history blame contribute delete
260 Bytes
# backend/model_loader.py
from transformers import AutoTokenizer, AutoModelForCausalLM
MODEL_PATH = "b1228032/nursing-bloom-1b7-full"
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
model = AutoModelForCausalLM.from_pretrained(MODEL_PATH)
model.eval()