Update Curator links
Browse files
README.md
CHANGED
|
@@ -38,7 +38,7 @@ The Internal State of an LLM Knows When It's Lying: https://arxiv.org/pdf/2304.1
|
|
| 38 |
|
| 39 |
## Software Integration:
|
| 40 |
**Runtime Engine(s):**
|
| 41 |
-
* NeMo Curator: https://github.com/NVIDIA/
|
| 42 |
* Aegis: https://huggingface.co/nvidia/Aegis-AI-Content-Safety-LlamaGuard-Defensive-1.0 <br>
|
| 43 |
|
| 44 |
**Supported Hardware Microarchitecture Compatibility:** <br>
|
|
@@ -74,8 +74,8 @@ Success is defined as having an acceptable catch rate (recall scores for each at
|
|
| 74 |
* A100 80GB GPU <br>
|
| 75 |
|
| 76 |
## How to Use in NeMo Curator:
|
| 77 |
-
The inference code is available on [NeMo Curator's GitHub repository](https://github.com/NVIDIA/
|
| 78 |
-
Check out [this example notebook](https://github.com/NVIDIA/
|
| 79 |
|
| 80 |
## How to Use in Transformers:
|
| 81 |
To use this AEGIS classifiers, you must get access to Llama Guard on Hugging Face here: https://huggingface.co/meta-llama/LlamaGuard-7b. Afterwards, you should set up a [user access token](https://huggingface.co/docs/hub/en/security-tokens) and pass that token into the constructor of this classifier.
|
|
|
|
| 38 |
|
| 39 |
## Software Integration:
|
| 40 |
**Runtime Engine(s):**
|
| 41 |
+
* NeMo Curator: https://github.com/NVIDIA-NeMo/Curator <br>
|
| 42 |
* Aegis: https://huggingface.co/nvidia/Aegis-AI-Content-Safety-LlamaGuard-Defensive-1.0 <br>
|
| 43 |
|
| 44 |
**Supported Hardware Microarchitecture Compatibility:** <br>
|
|
|
|
| 74 |
* A100 80GB GPU <br>
|
| 75 |
|
| 76 |
## How to Use in NeMo Curator:
|
| 77 |
+
The inference code is available on [NeMo Curator's GitHub repository](https://github.com/NVIDIA-NeMo/Curator). <br>
|
| 78 |
+
Check out [this example notebook](https://github.com/NVIDIA-NeMo/Curator/blob/main/tutorials/text/distributed-data-classification/instruction-data-guard-classification.ipynb) to get started.
|
| 79 |
|
| 80 |
## How to Use in Transformers:
|
| 81 |
To use this AEGIS classifiers, you must get access to Llama Guard on Hugging Face here: https://huggingface.co/meta-llama/LlamaGuard-7b. Afterwards, you should set up a [user access token](https://huggingface.co/docs/hub/en/security-tokens) and pass that token into the constructor of this classifier.
|