Spaces:
Sleeping
Sleeping
| # 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() |