@@ -1381,7 +1381,7 @@ def forward(self,
1381
1381
_bos_id = bos_token_id ,
1382
1382
_eos_id = eos_token_id ,
1383
1383
_max_out_len = max_out_len ,
1384
- _diversity_rate = diversity_rate ,
1384
+ _diversity_rate = - diversity_rate ,
1385
1385
_unk_id = self ._unk_id ,
1386
1386
_mask_id = self ._mask_id ,
1387
1387
_temperature = temperature ,
@@ -1625,7 +1625,7 @@ def forward(self,
1625
1625
self .linear_weight , self .linear_bias , self .pos_emb ,
1626
1626
decoding_strategy , beam_size , top_k , top_p , self ._n_head ,
1627
1627
int (self ._d_model / self ._n_head ), self ._num_decoder_layers ,
1628
- bos_token_id , eos_token_id , max_out_len , diversity_rate , rel_len ,
1628
+ bos_token_id , eos_token_id , max_out_len , - diversity_rate , rel_len ,
1629
1629
alpha , early_stopping )
1630
1630
1631
1631
ids = finalize (
@@ -1877,7 +1877,7 @@ def forward(self,
1877
1877
self .linear_bias , self .pos_emb , trg_word , decoding_strategy ,
1878
1878
beam_size , top_k , top_p , self ._n_head ,
1879
1879
int (self ._d_model / self ._n_head ), self ._num_decoder_layers ,
1880
- bos_token_id , eos_token_id , max_out_len , diversity_rate , rel_len ,
1880
+ bos_token_id , eos_token_id , max_out_len , - diversity_rate , rel_len ,
1881
1881
alpha , temperature , early_stopping , self ._hidden_act )
1882
1882
1883
1883
ids = finalize (
0 commit comments