Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4# Load the base model
5base_model_id = "mistralai/Mistral-7B-v0.1"
6tokenizer = AutoTokenizer.from_pretrained(base_model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 base_model_id,
9 device_map="auto",
10 load_in_8bit=True # Adjust based on your hardware
11)
12
13# Load the LoRA adapter
14adapter_model_id = "abhishekpawar7218/mistral-7b-lora-adapter"
15model = PeftModel.from_pretrained(model, adapter_model_id)
16
17# Generate text
18input_text = "Your prompt here"
19inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
20outputs = model.generate(**inputs, max_length=100)
21print(tokenizer.decode(outputs[0], skip_special_tokens=True))