Add PoC

README.md

@@ -1,8 +1,3 @@
 # PyTorch Lightning ACE PoC
 
-Multiple unsafe torch.load paths in PyTorch Lightning:
-
-1. deepspeed.py: explicit weights_only=False
-2. upgrade_checkpoint.py: bare torch.load without weights_only
-3. model_parallel.py: bare torch.load without weights_only
-4. xla.py / xla_fsdp.py: bare torch.load without weights_only
+lightning/pytorch/utilities/deepspeed.py:96,99 explicit weights_only=False.

malicious_lightning.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9e050af314d406c9f55015c3d9c7e282ee8e3cdc0bdbf47187ecd7d8a6b1b1f
+size 1789