tests/generation/test_generation_utils.py · 3b00b623b7cad9e1b7c71c97fff24a0286b37045 · chenpangpang / transformers

Fix `top_k_top_p_filtering` having unexpected behavior (#17744) · 3b00b623

unifyh authored Jun 22, 2022

- Fix `top_k_top_p_filtering` not passing `filter_value` to
   `TopPLogitsWarper` causing any top-p filtered logits to be -inf
   instead of specified value

 - Add corresponding test

3b00b623

test_generation_utils.py 113 KB

Replace test_generation_utils.py