Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = "meta-llama/Meta-Llama-3-8B"
5
6model = AutoModelForCausalLM.from_pretrained(base_model)
7model = PeftModel.from_pretrained(model, "Hiper-rex/mental-health-model")
8
9tokenizer = AutoTokenizer.from_pretrained(base_model)