diff --git a/modules/sampler_hijack.py b/modules/sampler_hijack.py index 97951dd2..24dbcf2e 100644 --- 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