Vashesh
/

Bert_Finetuned4

Text Classification

text-embeddings-inference

Model card Files Files and versions

Vashesh commited on Jul 5, 2024

Commit

c1ae553

·

verified ·

1 Parent(s): 33c1a52

Update config.json

Files changed (1) hide show

config.json +2 -0

config.json CHANGED Viewed

@@ -442,6 +442,8 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
   "transformers_version": "4.41.2",
   "type_vocab_size": 2,
   "use_cache": true,

   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
+  "problem_type": "single_label_classification",
+  "torch_dtype": "float32",
   "transformers_version": "4.41.2",
   "type_vocab_size": 2,
   "use_cache": true,