DataPilot
commited on
trancformers
Browse files
README.md
CHANGED
|
@@ -12,12 +12,13 @@ language: ja
|
|
| 12 |
|
| 13 |
```python
|
| 14 |
|
| 15 |
-
#pip install
|
| 16 |
#pip install pipline
|
| 17 |
#pip install protobu
|
| 18 |
#pip install accelerate
|
| 19 |
#pip install sentencepiece
|
| 20 |
-
#pip install torch
|
|
|
|
| 21 |
|
| 22 |
import torch
|
| 23 |
from transformers import AutoModelForSequenceClassification
|
|
|
|
| 12 |
|
| 13 |
```python
|
| 14 |
|
| 15 |
+
#pip install transformers
|
| 16 |
#pip install pipline
|
| 17 |
#pip install protobu
|
| 18 |
#pip install accelerate
|
| 19 |
#pip install sentencepiece
|
| 20 |
+
#pip install torch ← CPU推論で動作させる場合。
|
| 21 |
+
#pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 ← GPU推論で動作させる場合。
|
| 22 |
|
| 23 |
import torch
|
| 24 |
from transformers import AutoModelForSequenceClassification
|