Spaces:
Runtime error
Runtime error
split added
Browse files- src/models/model.py +1 -1
src/models/model.py
CHANGED
@@ -340,7 +340,7 @@ class Summarization:
|
|
340 |
else None
|
341 |
)
|
342 |
|
343 |
-
gpus = 1 if use_gpu else 0
|
344 |
|
345 |
trainer = Trainer(
|
346 |
logger=[MLlogger],
|
|
|
340 |
else None
|
341 |
)
|
342 |
|
343 |
+
gpus = -1 if use_gpu and torch.cuda.is_available() else 0
|
344 |
|
345 |
trainer = Trainer(
|
346 |
logger=[MLlogger],
|