Update src/forge/util/ops.py

Co-authored-by: Joe Cummings <[email protected]>

Author: felipemello1

src/forge/util/ops.py

@@ -79,7 +79,7 @@ def compute_logprobs(
 
     # get per-token log probs
     batch_size, seq_len, vocab_size = scaled_logits_fp32.shape
-    log_probs = -F.cross_entropy(
+    logprobs = -F.cross_entropy(
         scaled_logits_fp32.reshape(-1, vocab_size),
         input_ids.reshape(-1).long(),
         reduction="none",