Generate: add SequenceBiasLogitsProcessor (#24334)

2023-06-21 11:14:41 +01:00
parent 45f71d793d
commit 5f0801d174
8 changed files with 241 additions and 123 deletions
--- a/docs/source/en/internal/generation_utils.md
+++ b/docs/source/en/internal/generation_utils.md
@@ -141,6 +141,9 @@ generation.
 [[autodoc]] NoRepeatNGramLogitsProcessor
    - __call__

+[[autodoc]] SequenceBiasLogitsProcessor
+    - __call__
+
 [[autodoc]] NoBadWordsLogitsProcessor
    - __call__