Views
No views yet
meta-llama/Llama-3.1-8B-Instruct1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base = "meta-llama/Llama-3.1-8B-Instruct"
5adapter = "mattany/organic-dpo-3.1-8B-lora"
6
7tokenizer = AutoTokenizer.from_pretrained(base)
8model = AutoModelForCausalLM.from_pretrained(base, device_map="auto", torch_dtype="auto")
9model = PeftModel.from_pretrained(model, adapter)
10
11prompt = "What is photosynthesis?"
12inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
13outputs = model.generate(**inputs, max_new_tokens=256)
14print(tokenizer.decode(outputs[0], skip_special_tokens=True))