Update README.md

README.md

@@ -62,7 +62,8 @@ are simply reacting to it, which is how markets should operate."""
 import transformers
 import os
 import torch
-os.environ["CUDA_VISIBLE_DEVICES"]="5"
+#If you have more than one GPU, you can specify here which one to use
+os.environ["CUDA_VISIBLE_DEVICES"]="5" 
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(device)