Fix safety issue

Signed-off-by: Boxiang Wang <boxiangw@oci-hsg-cs-001-vscode-01.cm.cluster>

Author: Boxiang Wang

nemo/lightning/pytorch/strategies/megatron_strategy.py

@@ -1115,7 +1115,10 @@ def _save_fsdp_dtensor_common_state(self, state_dict, ckpt_dir):
         torch.save(state_dict, os.path.join(ckpt_dir, "common.pt"))
 
     def _load_fsdp_dtensor_common_state(self, ckpt_dir):
-        return torch.load(os.path.join(ckpt_dir, "common.pt"), weights_only=False)
+        # Note: set to weights_only=True for safety reason
+        # It should be switching to safetensor but NeMo is deprecated.
+        # Please use Megatron-Bridge for better checkpointing support.
+        return torch.load(os.path.join(ckpt_dir, "common.pt"))
 
     def _load_fsdp_dtensor_checkpoint(self, path, sharded_state_dict, strict):
         from torch.distributed.checkpoint import default_planner