From 56dd24434048d01a7eb34f470701dcf4e69d5d43 Mon Sep 17 00:00:00 2001 From: stevenxdavis <31548921+stevenxdavis@users.noreply.github.com> Date: Wed, 3 Sep 2025 06:39:12 -0500 Subject: [PATCH] Fix incorrect function call in test_qwen3_grpo.py (#3212) * 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 --------- Co-authored-by: Daniel Han --- blackwell/test_qwen3_grpo.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/blackwell/test_qwen3_grpo.py b/blackwell/test_qwen3_grpo.py index bf56ad338c..6232ceed17 100644 --- a/blackwell/test_qwen3_grpo.py +++ b/blackwell/test_qwen3_grpo.py @@ -415,7 +415,8 @@ sampling_params = SamplingParams( top_k=50, max_tokens=1024, ) -model.disable_gradient_checkpointing() + + output = ( model.fast_generate( [text],