LemiSt
/

code-segmentor-distilbert

Token Classification

Token Classification

Model card Files Files and versions

LemiSt commited on Jan 14, 2024

Commit

32c3b16

·

verified ·

1 Parent(s): b5ed404

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,8 +1,8 @@
 {
-  "_name_or_path": "code-segmentor-distilbert/checkpoint-55725",
   "activation": "gelu",
   "architectures": [
-    "DistilBertModel"
   ],
   "attention_dropout": 0.1,
   "dim": 768,

 {
+  "_name_or_path": "code-segmentor-distilbert",
   "activation": "gelu",
   "architectures": [
+    "DistilBertForTokenClassification"
   ],
   "attention_dropout": 0.1,
   "dim": 768,