mirror of
https://github.com/guillaume-be/rust-bert.git
synced 2024-09-11 12:55:34 +03:00
Updated generation benchmark configuration
This commit is contained in:
parent
56574a15d5
commit
822f0bf19f
@ -33,6 +33,8 @@ fn create_text_generation_model() -> TextGenerationModel {
|
||||
repetition_penalty: 1.0,
|
||||
length_penalty: 1.0,
|
||||
no_repeat_ngram_size: 3,
|
||||
num_beam_groups: None,
|
||||
diversity_penalty: None,
|
||||
num_return_sequences: 5,
|
||||
device: Device::cuda_if_available(),
|
||||
};
|
||||
|
Loading…
Reference in New Issue
Block a user