diff --git a/slime/backends/megatron_utils/model.py b/slime/backends/megatron_utils/model.py index dd3724424..baaf7157a 100644 --- a/slime/backends/megatron_utils/model.py +++ b/slime/backends/megatron_utils/model.py @@ -293,6 +293,10 @@ def forward_step( origin_values[origin_index] = value values = origin_values rollout_data[f"{store_prefix}{key}"] = values + + # 显式释放 forward_data_store 以避免显存泄漏 + del forward_data_store + return rollout_data