From cbfa7a20f9035f352597d56aa3249cd4120704a4 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 23 Dec 2025 05:42:58 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 4ffd790d1e..4ea36519d9 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -817,8 +817,8 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): " memory_gb_left = psutil.virtual_memory().available / (1024**3)\n" " if memory_gb_left <= 4: dataset_num_proc = 1 # Too risky, so set to 1\n" " elif memory_gb_left <= 6: dataset_num_proc = min(2, dataset_num_proc)\n" - " elif memory_gb_left <= 8: dataset_num_proc = min(4, dataset_num_proc)\n" - " elif memory_gb_left <= 12: dataset_num_proc = min(6, dataset_num_proc)\n" + " elif memory_gb_left <= 10: dataset_num_proc = min(4, dataset_num_proc)\n" + " elif memory_gb_left <= 14: dataset_num_proc = min(6, dataset_num_proc)\n" ) extra_args += num_proc_check