aliasgerovs commited on
Commit
1ff94f4
·
1 Parent(s): 7454788

updated GPU logic

Browse files
Files changed (1) hide show
  1. humanize.py +1 -1
humanize.py CHANGED
@@ -7,7 +7,7 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
7
 
8
  nltk.download("punkt")
9
  # autodetect the available device
10
- GPU_IDX = 0 # which GPU to use
11
  if torch.cuda.is_available():
12
  num_gpus = torch.cuda.device_count()
13
  print(f"Number of available GPUs: {num_gpus}")
 
7
 
8
  nltk.download("punkt")
9
  # autodetect the available device
10
+ GPU_IDX = 1 # which GPU to use
11
  if torch.cuda.is_available():
12
  num_gpus = torch.cuda.device_count()
13
  print(f"Number of available GPUs: {num_gpus}")