xglm

huggingface · Feb 6, 2025 · 5204b53 · 5204b53
1 parent 2f951dd
commit 5204b53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/models/xglm/modeling_xglm.py b/src/transformers/models/xglm/modeling_xglm.py
@@ -689,7 +689,7 @@ def forward(
         )
 
 
-def XGLMCrossEntropyLoss(
+def xglm_cross_entropy_loss(
     logits,
     labels,
     num_items_in_batch: int = None,
@@ -735,7 +735,7 @@ def __init__(self, config):
         # Initialize weights and apply final processing
         self.post_init()
 
-        self._loss_function = XGLMCrossEntropyLoss
+        self._loss_function = xglm_cross_entropy_loss
 
     def get_input_embeddings(self):
         return self.model.embed_tokens