This commit is contained in:
OlivierDehaene 2023-02-28 18:21:51 +01:00
parent 1064950462
commit 299f7367a5

View File

@ -234,6 +234,7 @@ mod tests {
do_sample: false, do_sample: false,
seed: 0, seed: 0,
repetition_penalty: 0.0, repetition_penalty: 0.0,
watermark: false
}, },
stopping_parameters: StoppingCriteriaParameters { stopping_parameters: StoppingCriteriaParameters {
max_new_tokens: 0, max_new_tokens: 0,