Update text_encoder/config.json

text_encoder/config.json

@@ -3,9 +3,9 @@
     "T5EncoderModel"
   ],
   "classifier_dropout": 0.0,
-  "d_ff": 10240,
+  "d_ff": 5120,
   "d_kv": 64,
-  "d_model": 4096,
+  "d_model": 2048,
   "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": "t5",
   "num_decoder_layers": 24,
-  "num_heads": 64,
+  "num_heads": 32,
   "num_layers": 24,
   "output_past": true,
   "pad_token_id": 0,
@@ -25,7 +25,7 @@
   "relative_attention_num_buckets": 32,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "4.41.2",
+  "transformers_version": "4.44.2",
   "use_cache": true,
   "vocab_size": 32128
 }