Update loss.py

Author: fzyzcjy

slime/backends/megatron_utils/loss.py

@@ -72,7 +72,9 @@ def get_log_probs_and_entropy(
     total_lengths: list[int],
     response_lengths: list[int],
     with_entropy: bool = False,
+    non_loss_data: bool = True,
 ) -> dict[str, list[torch.Tensor]]:
+    assert non_loss_data
     log_probs_list = []
     entropy_list = []
     for logits_chunk, tokens_chunk in get_responses(