Correct layer name for TFCTRL

huggingface · Jan 27, 2025 · 4620592 · 4620592
1 parent bb02fd1
commit 4620592
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/ctrl/modeling_tf_ctrl.py b/src/transformers/models/ctrl/modeling_tf_ctrl.py
@@ -869,7 +869,7 @@ def call(
             training=training,
         )
         hidden_states = transformer_outputs[0]
-        logits = self.score(hidden_states)
+        logits = self.classifier(hidden_states)
         logits_shape = shape_list(logits)
         batch_size = logits_shape[0]