Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load base model and tokenizer
5base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
6tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
7
8# Load LoRA adapters
9model = PeftModel.from_pretrained(base_model, "balazsthomay/llama-3.2-3b-motivational-quotes-lora")
10
11# Generate motivational advice
12prompt = "Give me advice about perseverance"
13inputs = tokenizer(prompt, return_tensors="pt")
14outputs = model.generate(**inputs, max_new_tokens=150)
15response = tokenizer.decode(outputs[0], skip_special_tokens=True)