Generate: unify LogitsWarper and LogitsProcessor (#32626)

2024-08-16 11:20:41 +01:00
parent 5fd7ca7bc9
commit 70d5df6107
20 changed files with 186 additions and 623 deletions
--- a/docs/source/en/internal/generation_utils.md
+++ b/docs/source/en/internal/generation_utils.md
@@ -158,9 +158,6 @@ generation.
 [[autodoc]] LogitsProcessorList
    - __call__

-[[autodoc]] LogitsWarper
-    - __call__
-
 [[autodoc]] MinLengthLogitsProcessor
    - __call__

@@ -421,4 +418,3 @@ A [`Constraint`] can be used to force the generation to include specific tokens

 [[autodoc]] WatermarkDetector
    - __call__
-