Views
No views yet
pip install transformers peft torch1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5# Load base model and tokenizer
6base_model_name = "mistralai/Mistral-7B-Instruct-v0.3"
7adapter_path = "Oliver1703dk/meal_review_fine_tuned_adapter_bigger"
8output_dir = "./merged_model"
9
10tokenizer = AutoTokenizer.from_pretrained(base_model_name)
11base_model = AutoModelForCausalLM.from_pretrained(
12 base_model_name,
13 torch_dtype=torch.float16,
14 device_map="auto"
15)
16
17# Load LoRA adapter
18model = PeftModel.from_pretrained(base_model, adapter_path)
19
20# Merge adapter with base model
21merged_model = model.merge_and_unload()
22
23# Save merged model (optional)
24merged_model.save_pretrained(output_dir)
25tokenizer.save_pretrained(output_dir)
26
27# Inference
28prompt = "Write a review for a delicious Italian meal."
29inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
30outputs = merged_model.generate(**inputs, max_length=200)
31print(tokenizer.decode(outputs[0], skip_special_tokens=True))