From 9c0afdaf7b091c341072b432ad6ee17ba7a5016b Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 20 Nov 2020 22:07:21 +0100 Subject: [PATCH] fix flaky ci (#8694) --- tests/test_generation_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_generation_utils.py b/tests/test_generation_utils.py index 433dad34e6..dee7873fb9 100644 --- a/tests/test_generation_utils.py +++ b/tests/test_generation_utils.py @@ -89,9 +89,9 @@ class GenerationTesterMixin: warp_kwargs = {"top_k": 10, "top_p": 0.7, "temperature": 0.7} logits_warper = LogitsProcessorList( [ + TemperatureLogitsWarper(warp_kwargs["temperature"]), TopKLogitsWarper(top_k=warp_kwargs["top_k"], min_tokens_to_keep=(2 if num_beams > 1 else 1)), TopPLogitsWarper(top_p=warp_kwargs["top_p"], min_tokens_to_keep=(2 if num_beams > 1 else 1)), - TemperatureLogitsWarper(warp_kwargs["temperature"]), ] ) return warp_kwargs, logits_warper