25.05.15:update modeling_dolly code and code comments

This commit is contained in:
coffee
2025-05-15 10:51:57 +08:00
parent 630d761f26
commit b48ff8a049
+1 -1
View File
@@ -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