Add Needle

Author: monoxgas

README.md

@@ -12,4 +12,8 @@ Implementation of ["Fast Adversarial Attacks on Language Models In One GPU Minut
 
 **[Llama PGD](scripts/pgd.py)**
 
-Implementation of ["Attacking Large Language Models with Projected Gradient Descent"](https://arxiv.org/abs/2402.09154) for Llama model variants with LitGPT. At teh time of release the authors have not posted any reference code, so be careful.
+Implementation of ["Attacking Large Language Models with Projected Gradient Descent"](https://arxiv.org/abs/2402.09154) for Llama model variants with LitGPT. At teh time of release the authors have not posted any reference code, so be careful.
+
+**Needle [Triage](notebooks/Needle%20-%20Triage.ipynb)/[Fix](notebooks/Needle%20-%20Fix.ipynb)**
+
+Research in partnership with [OpenSSF](https://openssf.org) for the [AIxCC Event](https://aicyberchallenge.com/).

notebooks/Mistral - Adversarial Suffix.ipynb

@@ -109,7 +109,7 @@
     "\n",
     "prompts: list[Prompt] = []\n",
     "\n",
-    "with open(\"prompts.json\", \"r\") as f:\n",
+    "with open(\"data/prompts.json\", \"r\") as f:\n",
     "    for name, content in json.load(f).items():\n",
     "        if (\n",
     "            len(content) < PROMPT_LEN_RANGE[0]\n",

notebooks/Mistral - BEAST Beam Attack.ipynb

@@ -111,7 +111,7 @@
     "\n",
     "prompts: list[Prompt] = []\n",
     "\n",
-    "with open(\"prompts.json\", \"r\") as f:\n",
+    "with open(\"data/prompts.json\", \"r\") as f:\n",
     "    for name, content in json.load(f).items():\n",
     "        if (\n",
     "            len(content) < PROMPT_LENGTHS[0]\n",