SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix flaky ci (#8694)

Browse Source
This commit is contained in:
Patrick von Platen
2020-11-20 22:07:21 +01:00
committed by GitHub
parent 29bdb88368
commit 9c0afdaf7b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_generation_utils.py
View File

@@ -89,9 +89,9 @@ class GenerationTesterMixin:
warp_kwargs = {"top_k": 10, "top_p": 0.7, "temperature": 0.7} warp_kwargs = {"top_k": 10, "top_p": 0.7, "temperature": 0.7}
logits_warper = LogitsProcessorList( 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)), 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)), 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 return warp_kwargs, logits_warper