Text2Text Generation
Transformers
PyTorch
t5
text-generation-inference
Inference Endpoints
yuewang-sf commited on
Commit
fe53abf
1 Parent(s): 71b98f1

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -7
config.json CHANGED
@@ -1,14 +1,14 @@
1
  {
2
- "_name_or_path": "/export/share/wang.y/workspace/large_codet5_pretrain/trainer_code/saved_models/codet5_large_e10_bs1920_lr1_warmup500_from_e150/checkpoint-90000",
3
  "architectures": [
4
- "T5ForConditionalGeneration"
5
  ],
 
6
  "d_ff": 4096,
7
  "d_kv": 64,
8
  "d_model": 1024,
9
  "decoder_start_token_id": 0,
10
  "dropout_rate": 0.1,
11
- "eos_token_id": 1,
12
  "feed_forward_proj": "relu",
13
  "initializer_factor": 1.0,
14
  "is_encoder_decoder": true,
@@ -20,8 +20,8 @@
20
  "num_layers": 24,
21
  "output_past": true,
22
  "pad_token_id": 0,
 
23
  "relative_attention_num_buckets": 32,
24
- "return_dict": false,
25
  "task_specific_params": {
26
  "summarization": {
27
  "early_stopping": true,
@@ -51,8 +51,7 @@
51
  "prefix": "translate English to Romanian: "
52
  }
53
  },
54
- "torch_dtype": "float16",
55
- "transformers_version": "4.16.2",
56
  "use_cache": true,
57
  "vocab_size": 32100
58
- }
 
1
  {
 
2
  "architectures": [
3
+ "T5WithLMHeadModel"
4
  ],
5
+ "bos_token_id": 1,
6
  "d_ff": 4096,
7
  "d_kv": 64,
8
  "d_model": 1024,
9
  "decoder_start_token_id": 0,
10
  "dropout_rate": 0.1,
11
+ "eos_token_id": 2,
12
  "feed_forward_proj": "relu",
13
  "initializer_factor": 1.0,
14
  "is_encoder_decoder": true,
 
20
  "num_layers": 24,
21
  "output_past": true,
22
  "pad_token_id": 0,
23
+ "relative_attention_max_distance": 128,
24
  "relative_attention_num_buckets": 32,
 
25
  "task_specific_params": {
26
  "summarization": {
27
  "early_stopping": true,
 
51
  "prefix": "translate English to Romanian: "
52
  }
53
  },
54
+ "transformers_version": "4.19.2",
 
55
  "use_cache": true,
56
  "vocab_size": 32100
57
+ }