Nicola De Cao 2f9d49b389 Adding PrefixConstrainedLogitsProcessor (#8529) ...

* Adding PrefixConstrainedLogitsProcessor

* fixing RAG and style_doc

* fixing black (v20 instead of v19)

* Improving doc in generation_logits_process.py

* Improving docs and typing in generation_utils.py

* docs improvement

* adding test and fixing doc typo

* fixing doc_len

* isort on test

* fixed test

* improve docstring a bit

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2020-11-18 17:06:25 +01:00

12 KiB

Raw Blame History

View Raw