> For the complete documentation index, see [llms.txt](https://vigil.deadbits.ai/llms.txt). Markdown versions of documentation pages are available by appending `.md` to page URLs; this page is available as [Markdown](https://vigil.deadbits.ai/overview/use-vigil/load-datasets.md).

# Load Datasets

If you don't intend to use the vector database scanner, you can skip this step.

Embeddings are currently available with three models, or you can bring your own dataset.

* `text-embedding-ada-002`
* `all-MiniLM-L6-v2`
* `all-mpnet-base-v2`

If there is a model you'd like to see added, feel free to [open a Github Issue](https://github.com/deadbits/vigil-llm/issues).

{% tabs %}
{% tab title="text-embedding-ada-002" %}

| Repo                                                                                                          |
| ------------------------------------------------------------------------------------------------------------- |
| [vigil-instruction-bypass-ada-002](https://huggingface.co/datasets/deadbits/vigil-instruction-bypass-ada-002) |
| [vigil-jailbreaks-ada-002](https://huggingface.co/datasets/deadbits/vigil-jailbreak-ada-002)                  |
| {% endtab %}                                                                                                  |

{% tab title="all-MiniLM-L6-v2" %}

| Repo                                                                                                                            |
| ------------------------------------------------------------------------------------------------------------------------------- |
| [vigil-instruction-bypass-all-MiniLM-L6-v2](https://huggingface.co/datasets/deadbits/vigil-instruction-bypass-all-MiniLM-L6-v2) |
| [vigil-jailbreaks-all-MiniLM-L6-v2](https://huggingface.co/datasets/deadbits/vigil-jailbreak-all-MiniLM-L6-v2)                  |
| {% endtab %}                                                                                                                    |

{% tab title="all-mpnet-base-v2" %}

| Repo                                                                                                                              |
| --------------------------------------------------------------------------------------------------------------------------------- |
| [vigil-instruction-bypass-all-mpnet-base-v2](https://huggingface.co/datasets/deadbits/vigil-instruction-bypass-all-mpnet-base-v2) |
| [vigil-jailbreak-all-mpnet-base-v2](https://huggingface.co/datasets/deadbits/vigil-jailbreak-all-mpnet-base-v2)                   |
| {% endtab %}                                                                                                                      |
| {% endtabs %}                                                                                                                     |

## **Run loader**

Load the appropriate datasets for your embedding model with the `loader.py` utility.

**Example: OpenAI datasets**

```bash
python loader.py --conf conf/server.conf --dataset deadbits/vigil-instruction-bypass-ada-002
python loader.py --conf conf/server.conf --dataset deadbits/vigil-jailbreak-ada-002
```

You can also load your own datasets from [Hugging Face Hub](https://huggingface.co/datasets) as long as you use the columns:

| Column     | Type         |
| ---------- | ------------ |
| text       | string       |
| embeddings | list\[float] |
| model      | string       |