admin/LLM-from-scratch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

25.05.15:update modeling_dolly code and code comments

Browse Source
This commit is contained in:
coffee
2025-05-15 10:51:57 +08:00
parent d52f33aa4a
commit 7e55276b43
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src_llm/modeling_dolly.py
+1 -1
View File
@@ -782,7 +782,7 @@ class DollyForSequenceClassification(DollyPreTrainedModel):
) )
class Dolly3ForTokenClassification(DollyPreTrainedModel): class DollyForTokenClassification(DollyPreTrainedModel):
def __init__(self, config): def __init__(self, config):
super().__init__(config) super().__init__(config)
self.num_labels = config.num_labels self.num_labels = config.num_labels