A fine-tuned version of
Qwen/Qwen2.5-0.5B-Instruct trained on cybersecurity instruction-response data using LoRA (Low-Rank Adaptation).
1from unsloth import FastLanguageModel
2
3model, tokenizer = FastLanguageModel.from_pretrained(
4 model_name = "Aaqisher667/qwen05b-cyber-finetuned",
5 max_seq_length = 512,
6 dtype = None,
7 load_in_4bit = True,
8)
9
10FastLanguageModel.for_inference(model)
1messages = [
2 {"role": "system", "content": "You are a cybersecurity expert assistant."},
3 {"role": "user", "content": "What is a SQL injection attack?"},
4]
5
6chat = tokenizer.apply_chat_template(
7 messages,
8 tokenize=False,
9 add_generation_prompt=True
10)
11
12inputs = tokenizer(chat, return_tensors="pt").to(model.device)
13
14import torch
15with torch.no_grad():
16 out = model.generate(
17 **inputs,
18 max_new_tokens = 256,
19 temperature = 0.7,
20 do_sample = True,
21 )
22
23response = tokenizer.decode(
24 out[0][inputs["input_ids"].shape[1]:],
25 skip_special_tokens=True
26)
27print(response)
Runs comfortably on a free Google Colab T4 GPU.