iproskurina commited on
Commit
29ef0ea
1 Parent(s): d5b9966

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -1,6 +1,7 @@
1
  {
 
2
  "architectures": [
3
- "PipelinedRobertaForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.3,
6
  "bos_token_id": 3,
@@ -18,7 +19,7 @@
18
  "num_hidden_layers": 4,
19
  "pad_token_id": 1,
20
  "position_embedding_type": "relative_key_query",
21
- "torch_dtype": "float16",
22
  "transformers_version": "4.20.1",
23
  "type_vocab_size": 1,
24
  "use_cache": true,
 
1
  {
2
+ "_name_or_path": "./roberta_bebeshka/",
3
  "architectures": [
4
+ "RobertaForMaskedLM"
5
  ],
6
  "attention_probs_dropout_prob": 0.3,
7
  "bos_token_id": 3,
 
19
  "num_hidden_layers": 4,
20
  "pad_token_id": 1,
21
  "position_embedding_type": "relative_key_query",
22
+ "torch_dtype": "float32",
23
  "transformers_version": "4.20.1",
24
  "type_vocab_size": 1,
25
  "use_cache": true,