From 7e55276b431700dcc2f6700878a94be4500ad858 Mon Sep 17 00:00:00 2001 From: coffee Date: Thu, 15 May 2025 10:51:57 +0800 Subject: [PATCH] 25.05.15:update modeling_dolly code and code comments --- src_llm/modeling_dolly.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src_llm/modeling_dolly.py b/src_llm/modeling_dolly.py index 5b10d69..416e94d 100755 --- a/src_llm/modeling_dolly.py +++ b/src_llm/modeling_dolly.py @@ -782,7 +782,7 @@ class DollyForSequenceClassification(DollyPreTrainedModel): ) -class Dolly3ForTokenClassification(DollyPreTrainedModel): +class DollyForTokenClassification(DollyPreTrainedModel): def __init__(self, config): super().__init__(config) self.num_labels = config.num_labels