git clone https://github.com/j-cb/adv_robust_mad cd adv_robust_mad/

manually copy the dataset parquet files https://www.oxen.ai/synapsecai/synthetic-prompt-injections/file/main/ (download with python requests somehow fails)

python prepare_environment.py

(enter huggingface token and hit enter)

python prompt_guard_AT.py