f

Author: zhiyuan1i

src/lightning/fabric/strategies/deepspeed.py

@@ -506,7 +506,7 @@ def load_checkpoint(
 
         optimzer_state_requested = any(isinstance(item, (Optimizer, DeepSpeedOptimizer)) for item in state.values())
 
-        if isinstance(self.accelerator, Accelerator) and self.accelerator.get_device() != "cpu": # type: ignore[union-attr]
+        if isinstance(self.accelerator, Accelerator) and self.accelerator.get_device() != "cpu":
             getattr(torch, self.root_device.type.split(':')[0]).empty_cache()
         else:
             torch.cuda.empty_cache()