Commit
·
fe2b41d
1
Parent(s):
dc3621b
Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,13 @@ license: cc-by-nc-sa-4.0
|
|
| 9 |
This model represents an ONNX-optimized version of the original [roberta_toxicity_classifier](https://huggingface.co/s-nlp/roberta_toxicity_classifier) model.
|
| 10 |
It has been specifically tailored for GPUs and may exhibit variations in performance when run on CPUs.
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
## How to use
|
| 13 |
```python
|
| 14 |
from transformers import AutoTokenizer
|
|
|
|
| 9 |
This model represents an ONNX-optimized version of the original [roberta_toxicity_classifier](https://huggingface.co/s-nlp/roberta_toxicity_classifier) model.
|
| 10 |
It has been specifically tailored for GPUs and may exhibit variations in performance when run on CPUs.
|
| 11 |
|
| 12 |
+
## Dependencies
|
| 13 |
+
|
| 14 |
+
Please install the following dependency before you begin working with the model:
|
| 15 |
+
```sh
|
| 16 |
+
pip install optimum[onnxruntime-gpu]
|
| 17 |
+
```
|
| 18 |
+
|
| 19 |
## How to use
|
| 20 |
```python
|
| 21 |
from transformers import AutoTokenizer
|