Merge remote-tracking branch 'refs/remotes/origin/dev' into dev

oobabooga · Oct 24, 2024 · 3a92fa5 · 3a92fa5
2 parents 8deea29 + e1061ba
commit 3a92fa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/sampler_hijack.py b/modules/sampler_hijack.py
@@ -454,7 +454,7 @@ def get_logits_processor_patch(self, **kwargs):
             )
 
         # Stuff we don't need
-        elif warpers[i].__class__.__name__ in ['SuppressTokensLogitsProcessor', 'RepetitionPenaltyLogitsProcessor']:
+        elif warpers[i].__class__.__name__ in ['RepetitionPenaltyLogitsProcessor']:
             del warpers[i]
 
     # Add custom warpers