Upload MT5ForConditionalGeneration

config.json

@@ -1,11 +1,11 @@
 {
-  "_name_or_path": "hf-internal-testing/tiny-random-mt5",
+  "_name_or_path": "stas/mt5-tiny-random",
   "architectures": [
     "MT5ForConditionalGeneration"
   ],
-  "d_ff": 1024,
-  "d_kv": 64,
-  "d_model": 512,
+  "d_ff": 256,
+  "d_kv": 8,
+  "d_model": 64,
   "decoder_start_token_id": 0,
   "dense_act_fn": "gelu_new",
   "dropout_rate": 0.1,
@@ -17,7 +17,7 @@
   "layer_norm_epsilon": 1e-06,
   "model_type": "mt5",
   "num_decoder_layers": 8,
-  "num_heads": 6,
+  "num_heads": 4,
   "num_layers": 8,
   "pad_token_id": 0,
   "relative_attention_max_distance": 128,
@@ -27,5 +27,5 @@
   "torch_dtype": "float32",
   "transformers_version": "4.25.1",
   "use_cache": true,
-  "vocab_size": 250112
+  "vocab_size": 5100
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3b5cfaaea70c0479100da88500fe171ebc9905e4e2e251fdef1432a1f99750e4
-size 1200769797
+oid sha256:e116c4736d8cfebff79fecd6a86b6d72067d5bfc0e4a26cb3b201340a1979588
+size 6617157