From fb7c288c94d37cc2a19eb2086234d9215cd18e6f Mon Sep 17 00:00:00 2001 From: AdilZouitine Date: Tue, 29 Apr 2025 18:23:51 +0200 Subject: [PATCH] Update torch.load calls in network_utils.py to include weights_only=False, to ensure no regression with torch 2.6 update --- lerobot/scripts/server/network_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/scripts/server/network_utils.py b/lerobot/scripts/server/network_utils.py index b24c8f172..c62f6cbd9 100644 --- a/lerobot/scripts/server/network_utils.py +++ b/lerobot/scripts/server/network_utils.py @@ -111,7 +111,7 @@ def state_to_bytes(state_dict: dict[str, torch.Tensor]) -> bytes: def bytes_to_state_dict(buffer: bytes) -> dict[str, torch.Tensor]: buffer = io.BytesIO(buffer) buffer.seek(0) - return torch.load(buffer) # nosec B614: Safe usage of torch.load + return torch.load(buffer, weights_only=False) # nosec B614: Safe usage of torch.load def python_object_to_bytes(python_object: Any) -> bytes: @@ -129,7 +129,7 @@ def bytes_to_python_object(buffer: bytes) -> Any: def bytes_to_transitions(buffer: bytes) -> list[Transition]: buffer = io.BytesIO(buffer) buffer.seek(0) - transitions = torch.load(buffer) # nosec B614: Safe usage of torch.load + transitions = torch.load(buffer, weights_only=False) # nosec B614: Safe usage of torch.load # Add validation checks here return transitions