Views
No views yet
mistralai/Mistral-7B-Instruct-v0.2transformers, peft, and torch libraries installed.1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5base_model_id = "mistralai/Mistral-7B-Instruct-v0.2"
6lora_adapter_id = "Bharath5626/devops-lora-mistral" # Update with your specific HF path
7
8tokenizer = AutoTokenizer.from_pretrained(base_model_id)
9model = AutoModelForCausalLM.from_pretrained(
10 base_model_id,
11 torch_dtype=torch.float16,
12 device_map="auto"
13)
14model = PeftModel.from_pretrained(model, lora_adapter_id)
15
16# Inference Example
17instruction = "Fix docker container exits immediately"
18prompt = f"### Instruction:\n{instruction}\n\n### Response:\n"
19
20inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
21outputs = model.generate(**inputs, max_new_tokens=150)
22print(tokenizer.decode(outputs[0], skip_special_tokens=True))