Update model.py

src/models/model.py

@@ -348,6 +348,7 @@ class Summarization:
             else None
         )
 
+        gpus = -1 if use_gpu and torch.cuda.is_available() else 0
 
         trainer = Trainer(
             logger=[logger, MLlogger],