From 8ccde63de60c35a5ae6f801dc38552a2b8f74262 Mon Sep 17 00:00:00 2001 From: Matt Date: Mon, 27 Jan 2025 17:27:19 +0000 Subject: [PATCH] Correct layer name for TFCTRL --- src/transformers/models/ctrl/modeling_tf_ctrl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/ctrl/modeling_tf_ctrl.py b/src/transformers/models/ctrl/modeling_tf_ctrl.py index 2e72c417d503..8f35076ff949 100644 --- 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]