Spaces:
Runtime error
Runtime error
aliasgerovs
commited on
Commit
·
1ff94f4
1
Parent(s):
7454788
updated GPU logic
Browse files- 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 =
|
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}")
|