Update fsdp.py

Lightning-AI · Sep 11, 2024 · e3f530a · e3f530a
1 parent b04a0ec
commit e3f530a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/lightning_habana/pytorch/strategies/fsdp.py b/src/lightning_habana/pytorch/strategies/fsdp.py
@@ -153,7 +153,6 @@ def precision_plugin(self, precision_plugin: Optional[HPUFSDPPrecision]) -> None
     @override
     def root_device(self) -> torch.device:
         assert self.parallel_devices is not None
-        # HPUParallelStrategy.setup_hccl_env(self)
         return torch.device("hpu", torch.hpu.current_device())
 
     def _setup_model(self, model: Module) -> Module: