skip assigning Metric.process_group

NVIDIA · Jan 16, 2025 · 070d57e · 070d57e
1 parent 1c7f9a9
commit 070d57e
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/sub-packages/bionemo-llm/src/bionemo/llm/lightning.py b/sub-packages/bionemo-llm/src/bionemo/llm/lightning.py
@@ -315,10 +315,6 @@ def configure_model(self) -> None:
 
         # configure logger
         self.set_logging_group()
-        if self.train_ppl is not None:
-            self.train_ppl.process_group = self.logging_group
-        if self.valid_ppl is not None:
-            self.valid_ppl.process_group = self.logging_group
 
     def is_on_logging_device(self):
         return parallel_state.is_pipeline_last_stage() and parallel_state.get_tensor_model_parallel_rank() == 0