ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use regex instead of static string

Browse source
This commit is contained in:
Datta Nimmaturi 2025-12-08 05:50:02 +00:00
commit f5caa53e26
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
unsloth/models/rl_replacements.py
View file

@ -224,7 +224,13 @@ def grpo_trainer__generate_single_turn(function_name, function):
return function
# Remove the reload_weights collective RPC call from the generate function's source
function = function.replace('self.llm.collective_rpc("reload_weights")', "")
# function = function.replace('self.llm.collective_rpc("reload_weights")', "")
# The regex below does the same thing but is more flexible and can handle single or double quotes
function = re.sub(
r"self\.llm\.collective_rpc\(\s*(['\"])reload_weights\1\s*\)",
"",
function,
)
return function