Views
No views yet
microsoft/Phi-3-mini-4k-instruct using QLoRA on documentation
for a WiFi threat detection system. It can:1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load base model
5model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
6model = PeftModel.from_pretrained(model, "alantandrea/wifi-threat-detection-assistant")
7tokenizer = AutoTokenizer.from_pretrained("alantandrea/wifi-threat-detection-assistant")
8
9# Generate response
10prompt = "<|system|>\nYou are a WiFi security expert.<|end|>\n<|user|>\nWhat is a deauth attack?<|end|>\n<|assistant|>\n"
11inputs = tokenizer(prompt, return_tensors="pt")
12outputs = model.generate(**inputs, max_new_tokens=256)
13print(tokenizer.decode(outputs[0]))