Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("YOUR_USERNAME/dolphin-cybersec")
4tokenizer = AutoTokenizer.from_pretrained("YOUR_USERNAME/dolphin-cybersec")
5
6prompt = """Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
7
8### Instruction:
9What is SQL injection?
10
11### Input:
12
13
14### Response:
15"""
16
17inputs = tokenizer(prompt, return_tensors="pt")
18outputs = model.generate(**inputs, max_new_tokens=512)
19print(tokenizer.decode(outputs[0]))