momo commited on
Commit
3d298a6
1 Parent(s): 689f8f8
Files changed (1) hide show
  1. config.json +5 -4
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/home/suraj/projects/mbart-50/hf_models/mbart-50-large",
3
  "_num_labels": 3,
4
  "activation_dropout": 0.0,
5
  "activation_function": "gelu",
@@ -50,8 +50,9 @@
50
  "pad_token_id": 1,
51
  "scale_embedding": true,
52
  "static_position_embeddings": false,
53
- "transformers_version": "4.4.0.dev0",
 
 
54
  "use_cache": true,
55
- "vocab_size": 250054,
56
- "tokenizer_class": "MBart50Tokenizer"
57
  }
 
1
  {
2
+ "_name_or_path": "facebook/mbart-large-50",
3
  "_num_labels": 3,
4
  "activation_dropout": 0.0,
5
  "activation_function": "gelu",
 
50
  "pad_token_id": 1,
51
  "scale_embedding": true,
52
  "static_position_embeddings": false,
53
+ "tokenizer_class": "MBart50Tokenizer",
54
+ "torch_dtype": "float16",
55
+ "transformers_version": "4.20.0",
56
  "use_cache": true,
57
+ "vocab_size": 250054
 
58
  }