GleghornLab
/

cdsBERT-plus

Feature Extraction

protein language model

Inference Endpoints

Model card Files Files and versions Community

lhallee commited on Sep 13, 2023

Commit

0e1cee2

•

1 Parent(s): 0d1ee18

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -17,7 +17,7 @@
   "num_hidden_layers": 30,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
-  "torch_dtype": "float32",
   "transformers_version": "4.29.2",
   "type_vocab_size": 2,
   "use_cache": true,

   "num_hidden_layers": 30,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.29.2",
   "type_vocab_size": 2,
   "use_cache": true,