m-a-p/CodeFeedback-Filtered-Instruction
Viewer • Updated • 157k • 22.8k • 225
How to use nehalkamal7/devguard-qwen-1.5b-security with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-Coder-1.5B-Instruct")
model = PeftModel.from_pretrained(base_model, "nehalkamal7/devguard-qwen-1.5b-security")DevGuard-Qwen-1.5B-Security is a parameter-efficient fine-tuned (PEFT/QLoRA) adapter built on top of Qwen/Qwen2.5-Coder-1.5B-Instruct. It is specifically adapted to act as an automated code security auditor, identifying critical software vulnerabilities (such as injection vectors, deserialization risks, and dynamic evaluation flaws) and generating secure code remediations.
This model serves as the specialized domain SLM for the DevGuard AI platform.
Qwen/Qwen2.5-Coder-1.5B-Instructbitsandbytes and peft)You can load this model along with its base weights using Hugging Face transformers and peft:
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
base_model_id = "Qwen/Qwen2.5-Coder-1.5B-Instruct"
adapter_id = "nehalkamal7/devguard-qwen-1.5b-security"
# Load Tokenizer
tokenizer = AutoTokenizer.from_pretrained(base_model_id, trust_remote_code=True)
# Load 4-bit Base Model
model = AutoModelForCausalLM.from_pretrained(
base_model_id,
torch_dtype=torch.float16,
device_map="auto"
)
# Load DevGuard LoRA Adapter
model = PeftModel.from_pretrained(model, adapter_id)
# Perform Security Audit
prompt = """<|im_start|>system
You are DevGuard AI, an expert code security auditor.<|im_end|>
<|im_start|>user
Audit this code and provide secure remediation:
def execute_query(user_id, cursor):
query = "SELECT * FROM users WHERE id = " + user_id
cursor.execute(query)
<|im_end|>
<|im_start|>assistant
"""
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=300, temperature=0.2)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
Base model
Qwen/Qwen2.5-1.5B