sgugger
/

my-bert-model

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

sgugger commited on Sep 3, 2021

Commit

f6b867d

·

1 Parent(s): 1ddc92c

Add auto_map field

Files changed (1) hide show

config.json +3 -0

config.json CHANGED Viewed

@@ -4,6 +4,9 @@
     "MyBertModel"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
   "gradient_checkpointing": false,
   "hidden_act": "gelu",

     "MyBertModel"
   ],
   "attention_probs_dropout_prob": 0.1,
+  "auto_map": {
+    "AutoModel": "modeling.MyBertModel"
+  },
   "classifier_dropout": null,
   "gradient_checkpointing": false,
   "hidden_act": "gelu",