Update README.md
Browse files
README.md
CHANGED
|
@@ -24,7 +24,6 @@ Apart from this, it has been trained exactly as described in the tech report.
|
|
| 24 |
|
| 25 |
## Demo Code
|
| 26 |
```python
|
| 27 |
-
# export ACCELERATE_MIXED_PRECISION=bf16
|
| 28 |
import torch
|
| 29 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 30 |
device = "cuda" # you can use "auto" for placing the model on several GPUs if your GPUs can not fit the model
|
|
|
|
| 24 |
|
| 25 |
## Demo Code
|
| 26 |
```python
|
|
|
|
| 27 |
import torch
|
| 28 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 29 |
device = "cuda" # you can use "auto" for placing the model on several GPUs if your GPUs can not fit the model
|