Commit
·
6e540cf
1
Parent(s):
165cdba
Update README.md
Browse files
README.md
CHANGED
|
@@ -62,6 +62,7 @@ are simply reacting to it, which is how markets should operate."""
|
|
| 62 |
import transformers
|
| 63 |
import os
|
| 64 |
import torch
|
|
|
|
| 65 |
os.environ["CUDA_VISIBLE_DEVICES"]="0"
|
| 66 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 67 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 62 |
import transformers
|
| 63 |
import os
|
| 64 |
import torch
|
| 65 |
+
#If you have more than one GPU, you can specify here which one to use
|
| 66 |
os.environ["CUDA_VISIBLE_DEVICES"]="0"
|
| 67 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 68 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|