tests/generation/test_generation_logits_process.py · f7196f2e63b14e9fbb4ad664e71912aab3b484cf · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Fix decoding score comparison when using logits processors or warpers (#10638) · f7196f2e
Santiago Castro authored Apr 13, 2022
```
* Normalize using a logits warper

* Add a flag in `generate` to support the logit renormalization

* Add in RAG
```
f7196f2e

test_generation_logits_process.py 22.9 KB