Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
b48d762
1
Parent(s):
1df2d7b
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -348,6 +348,7 @@ class Summarization:
|
|
348 |
else None
|
349 |
)
|
350 |
|
|
|
351 |
|
352 |
trainer = Trainer(
|
353 |
logger=[logger, MLlogger],
|
|
|
348 |
else None
|
349 |
)
|
350 |
|
351 |
+
gpus = -1 if use_gpu and torch.cuda.is_available() else 0
|
352 |
|
353 |
trainer = Trainer(
|
354 |
logger=[logger, MLlogger],
|