Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
96f677f
1
Parent(s):
5866066
Update model.py
Browse files- src/models/model.py +3 -1
src/models/model.py
CHANGED
@@ -414,7 +414,9 @@ class Summarization:
|
|
414 |
if torch.cuda.is_available():
|
415 |
self.device = torch.device("cuda")
|
416 |
else:
|
417 |
-
raise Exception(
|
|
|
|
|
418 |
else:
|
419 |
self.device = torch.device("cpu")
|
420 |
|
|
|
414 |
if torch.cuda.is_available():
|
415 |
self.device = torch.device("cuda")
|
416 |
else:
|
417 |
+
raise Exception(
|
418 |
+
"exception ---> no gpu found. set use_gpu=False, to use CPU"
|
419 |
+
)
|
420 |
else:
|
421 |
self.device = torch.device("cpu")
|
422 |
|