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],