Raziel1234 commited on
Commit
b6134bb
·
verified ·
1 Parent(s): 767e324

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -2,6 +2,10 @@
2
  "architectures": [
3
  "OSTLMModel"
4
  ],
 
 
 
 
5
  "d_model": 512,
6
  "dim_feedforward": 2048,
7
  "dropout": 0.1,
@@ -12,6 +16,8 @@
12
  "num_decoder_layers": 6,
13
  "num_encoder_layers": 6,
14
  "pad_token_id": 0,
 
 
15
  "transformers_version": "4.56.0",
16
  "vocab_size": 119547
17
  }
 
2
  "architectures": [
3
  "OSTLMModel"
4
  ],
5
+ "auto_map": {
6
+ "AutoConfig": "configuration_ostlm.OSTLMConfig",
7
+ "AutoModelForSeq2SeqLM": "modeling_ostlm.OSTLMModel"
8
+ },
9
  "d_model": 512,
10
  "dim_feedforward": 2048,
11
  "dropout": 0.1,
 
16
  "num_decoder_layers": 6,
17
  "num_encoder_layers": 6,
18
  "pad_token_id": 0,
19
+ "decoder_start_token_id": 101,
20
+ "eos_token_id": 102,
21
  "transformers_version": "4.56.0",
22
  "vocab_size": 119547
23
  }