From ef8ff6f5c8fa8617c8e4952952e74d24c09875ae Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Sun, 15 Feb 2026 13:41:37 +0100 Subject: [PATCH] fix(ReforceXY): exclude gpu_memory_fraction from model params --- ReforceXY/user_data/freqaimodels/ReforceXY.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index b62f707..d003a3a 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -760,6 +760,8 @@ class ReforceXY(BaseReinforcementLearningModel): ) # "adamw" ) + model_params.pop("gpu_memory_fraction", None) + self._model_params_cache = model_params return copy.deepcopy(self._model_params_cache) -- 2.53.0