From f498c78d76584f2d618269f0e1a6864fa64de2cf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Miguel=20Gonz=C3=A1lez=20Duque?= Date: Sun, 2 Feb 2025 10:55:37 +0100 Subject: [PATCH] adds explicit weights_only flag in model loading --- .../solvers/bayesian_optimization/lambo2/solver.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py b/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py index c0815a6..1b56138 100644 --- a/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py +++ b/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py @@ -213,6 +213,7 @@ def _train_model_with_history( torch.load( load_checkpoint_from, map_location="cpu", + weights_only=False, )["state_dict"] )