Magic-AI
/

Magical-1-Star

Text Classification

Inference Endpoints

Model card Files Files and versions Community

underwater45 commited on Jul 6

Commit

1919dcb

•

1 Parent(s): f341ff7

Update config.json

Files changed (1) hide show

config.json +8 -5

config.json CHANGED Viewed

@@ -1,11 +1,14 @@
 {
     "model_name": "Magical-1 Star",
-    "model_type": "bert",
     "version": "1.0.0",
-    "bert_model": "bert-base-uncased",
-    "max_seq_length": 128,
-    "hidden_size": 768,
-    "num_classes": 2,
     "training": {
         "batch_size": 32,
         "num_epochs": 5,

 {
     "model_name": "Magical-1 Star",
     "version": "1.0.0",
+    "sentence_transformer": {
+        "model_name": "all-MiniLM-L6-v2",
+        "max_seq_length": 128
+    },
+    "classifier": {
+        "input_size": 384,
+        "num_classes": 2
+    },
     "training": {
         "batch_size": 32,
         "num_epochs": 5,