johngiorgi commited on
Commit
d7347c3
1 Parent(s): c3ddc98

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -45,6 +45,7 @@
45
  "max_decoder_position_embeddings": 1024,
46
  "max_encoder_position_embeddings": 16384,
47
  "model_type": "led",
 
48
  "num_hidden_layers": 6,
49
  "pad_token_id": 1,
50
  "torch_dtype": "float32",
 
45
  "max_decoder_position_embeddings": 1024,
46
  "max_encoder_position_embeddings": 16384,
47
  "model_type": "led",
48
+ "num_beams": 2,
49
  "num_hidden_layers": 6,
50
  "pad_token_id": 1,
51
  "torch_dtype": "float32",