Views
No views yet
1from unsloth import FastLanguageModel
2
3model, tokenizer = FastLanguageModel.from_pretrained(
4 model_name = "jartechsai/cybersec-mistral-7b",
5 max_seq_length = 2048,
6 dtype = None,
7 load_in_4bit = True,
8)
9
10FastLanguageModel.for_inference(model)
11
12alpaca_prompt = """Below is a cybersecurity question or task. Provide a thorough, accurate response.
13
14### Instruction:
15{}
16
17### Input:
18{}
19
20### Response:
21{}"""
22
23inputs = tokenizer([alpaca_prompt.format("What is a KARMA attack?", "", "")], return_tensors="pt").to("cuda")
24outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.3, do_sample=True, repetition_penalty=1.3)
25print(tokenizer.decode(outputs[0], skip_special_tokens=True))