set weights_only=False for load_debug_rollout_data (THUDM#680)

Author: zhuzilin

slime/ray/rollout.py

@@ -129,6 +129,7 @@ def _get_rollout_data(self, rollout_id):
         if self.args.load_debug_rollout_data:
             data = torch.load(
                 open(self.args.load_debug_rollout_data.format(rollout_id=rollout_id), "rb"),
+                weights_only=False,
             )["samples"]
             data = [Sample.from_dict(sample) for sample in data]
             if (ratio := self.args.load_debug_rollout_data_subsample) is not None: