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

Fix incorrect function call in test_qwen3_grpo.py (#3212)

Browse source 
* Update test_qwen3_grpo.py to correct function call

This test file uses the incorrect name for the function, which is gradient_checkpointing_disable(), not disable_gradient_checkpointing(). 

I copied the line from test_llama32_sft.py - I'm not sure if this actually is required, just wanted it consistent for when other people like me test this and have no clue what they're doing when it throws an exception.

* Update blackwell/test_qwen3_grpo.py

Co-authored-by: Daniel Han <danielhanchen@gmail.com>

---------

Co-authored-by: Daniel Han <danielhanchen@gmail.com>
This commit is contained in:
stevenxdavis 2025-09-03 06:39:12 -05:00 • committed by GitHub
commit 56dd244340
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
blackwell/test_qwen3_grpo.py
View file

@ -415,7 +415,8 @@ sampling_params = SamplingParams(
top_k=50,
max_tokens=1024,
)
model.disable_gradient_checkpointing()
output = (
model.fast_generate(
[text],