The Power of Words: Generating PowerShell Attacks from ...
Collection
4 items โข Updated โข 2
How to use dessertlab/offensive-powershell-codet5p-220m-py with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("dessertlab/offensive-powershell-codet5p-220m-py")
model = AutoModelForSeq2SeqLM.from_pretrained("dessertlab/offensive-powershell-codet5p-220m-py", device_map="auto")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("dessertlab/offensive-powershell-codet5p-220m-py") model = AutoModelForSeq2SeqLM.from_pretrained("dessertlab/offensive-powershell-codet5p-220m-py", device_map="auto")