Update README.md

README.md

@@ -61,6 +61,7 @@ are simply reacting to it, which is how markets should operate."""
 import transformers
 import os
 import torch
+#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")