fix flaky ci (#8694)

9c0afdaf · Patrick von Platen · GitHub · 29bdb883 · 9c0afdaf
Unverified Commit 9c0afdaf authored Nov 20, 2020 by Patrick von Platen Committed by GitHub Nov 20, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_generation_utils.py tests/test_generation_utils.py +1 -1

No files found.
--- 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