harsh-deepchecks
commited on
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
|