esuriddick commited on
Commit
e02d90c
1 Parent(s): 04d5c0d

Final config

Browse files
Files changed (1) hide show
  1. config.json +5 -0
config.json CHANGED
@@ -43,9 +43,14 @@
43
  "LABEL_1": 1,
44
  "LABEL_2": 2
45
  },
 
46
  "max_decoder_position_embeddings": 1024,
47
  "max_encoder_position_embeddings": 16384,
 
 
48
  "model_type": "led",
 
 
49
  "num_hidden_layers": 6,
50
  "pad_token_id": 1,
51
  "torch_dtype": "float32",
 
43
  "LABEL_1": 1,
44
  "LABEL_2": 2
45
  },
46
+ "length_penalty": 2,
47
  "max_decoder_position_embeddings": 1024,
48
  "max_encoder_position_embeddings": 16384,
49
+ "min_new_tokens": 100,
50
+ "max_new_tokens": 1024,
51
  "model_type": "led",
52
+ "no_repeat_ngram_size": 3,
53
+ "num_beams": 5,
54
  "num_hidden_layers": 6,
55
  "pad_token_id": 1,
56
  "torch_dtype": "float32",