Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.1-8B")
5model = PeftModel.from_pretrained(base, "Rudraksh225/llama-3.1-8b-mental-support")
6tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.1-8B")