rawsh
/

ms-marco-TinyBERT-L-2-ONNX

Model card Files Files and versions Community

rawsh commited on Sep 13

Commit

1a12515

•

1 Parent(s): eb1c6ba

Update config.json

Files changed (1) hide show

config.json +1 -0

config.json CHANGED Viewed

@@ -17,6 +17,7 @@
   "label2id": {
     "LABEL_0": 0
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",

   "label2id": {
     "LABEL_0": 0
   },
+  "torch_dtype": "float16",
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",