diff --git a/gemma/gm/losses/_npo.py b/gemma/gm/losses/_npo.py
index 0af58371..9976ccb4 100644
--- a/gemma/gm/losses/_npo.py
+++ b/gemma/gm/losses/_npo.py
@@ -30,7 +30,6 @@ class NpoLoss(kd.losses.Loss):
 
   Attributes:
     tau: The temperature of the loss.
-    label_smoothing: The label smoothing to apply to the loss.
     tokens: The key to the tokens to predict.
     sequence_mask: The key to the sequence mask.
     policy_logits: The key to the policy logits.