wangyulong commited on
Commit
b7b290d
•
1 Parent(s): 8429dcf

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -25,7 +25,7 @@
25
  "pooler_size_per_head": 128,
26
  "pooler_type": "first_token_transform",
27
  "position_embedding_type": "absolute",
28
- "torch_dtype": "float16",
29
  "transformers_version": "4.9.2",
30
  "type_vocab_size": 2,
31
  "use_cache": true,
 
25
  "pooler_size_per_head": 128,
26
  "pooler_type": "first_token_transform",
27
  "position_embedding_type": "absolute",
28
+ "torch_dtype": "float32",
29
  "transformers_version": "4.9.2",
30
  "type_vocab_size": 2,
31
  "use_cache": true,