Update rl.py

2025-06-21 23:29:08 -07:00 · 2025-06-21 23:29:08 -07:00 · c7c6f2c88d
commit c7c6f2c88d
parent d9be145849
1 changed files with 2 additions and 0 deletions
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@ -476,6 +476,8 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
        "top_k"                       : None,
        "vllm_mode"                   : "colocate",
        "generation_kwargs"           : {},
+        "bf16"                        : False,
+        "fp16"                        : False,
    }
    for k, v in replacements.items():
        x = f"{k}( = [^,\n]{{1,}})?,\n"