omidvaramin
/

Ht5-small

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

omidvaramin commited on Sep 28, 2023

Commit

9da993f

·

1 Parent(s): 6e540cf

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -63,7 +63,7 @@ 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"]="0"
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(device)

 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")
 print(device)