Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("path/to/model")
4model = AutoModelForSeq2SeqLM.from_pretrained("path/to/model")
5
6def generate_command(instruction):
7 text = f"Generate Linux command: {instruction}"
8 inputs = tokenizer(text, return_tensors="pt", max_length=128, truncation=True)
9 outputs = model.generate(**inputs, max_length=128, num_beams=4)
10 return tokenizer.decode(outputs[0], skip_special_tokens=True)
11
12print(generate_command("List all files"))