from peft import PeftModelfrom transformers import AutoModelForCausalLM, AutoTokenizer base = AutoModelForCausalLM.from_pretrained( "Qwen/Qwen2.5-Coder-7B-Instruct", dtype="bfloat16", device_map="auto")model = PeftModel.from_pretrained(base, "akeakeaki/pytorch-rag-lora-synth-v2")tok = AutoTokenizer.from_pretrained("akeakeaki/pytorch-rag-lora-synth-v2") messages = [ {"role": "system", "content": "You are an expert PyTorch assistant. " "Answer using ONLY the provided Context."}, {"role": "user", "content": "Context:\n<retrieved chunks>\n\nQuestion: <q>"},]inputs = tok.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True).to(model.device)out = model.generate(inputs, max_new_tokens=400)print(tok.decode(out[0][inputs.shape[1]:], skip_special_tokens=True))