agaresd
/

GEN-AI-Final-project

Text Classification

Model card Files Files and versions

agaresd commited on Nov 21, 2024

Commit

0790d80

·

verified ·

1 Parent(s): 5baaf72

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "architectures": [
     "CustomDialogTransformer"
   ],
-  "model_type": "bert",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "intermediate_size": 3072,

   "architectures": [
     "CustomDialogTransformer"
   ],
+  "model_type": "custom",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "intermediate_size": 3072,