switch to weights_only=True in filesystem.py

awaelchli · awaelchli · commit e0275f87e508 · 2024-07-15T20:47:23.000+02:00
diff --git a/torch/distributed/checkpoint/filesystem.py b/torch/distributed/checkpoint/filesystem.py
@@ -657,7 +657,7 @@ def read_data(self, plan: LoadPlan, planner: LoadPlanner) -> Future[None]:
                             torch.load(
                                 cast(IO[bytes], file_slice),
                                 map_location="cpu",
-                                weights_only=False
+                                weights_only=True,
                             ),
                         )
                         tensor = narrow_tensor_by_index(
