fix: restrict lightning module to only use 1 GPU for predict v3

embeddings/model/lightning_module/lightning_module.py

@@ -106,7 +106,7 @@ def _predict_with_trainer(
                     return_predictions=True,
                 )
         else:
-            raise RuntimeError("Got `False` for `trainer.is_global_zero` attribute!")
+            return
 
     def on_train_epoch_end(self) -> None:
         self._aggregate_and_log_metrics(self.train_metrics)