From 9dbcb1aeeaedd13a4f3e9718d8f567744a561e4a Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 27 Jun 2024 17:05:29 -0700
Subject: [PATCH] Small fix to make transformers 4.42 functional

---
 modules/sampler_hijack.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/sampler_hijack.py b/modules/sampler_hijack.py
index ad74d658..9fb661ae 100644
--- a/modules/sampler_hijack.py
+++ b/modules/sampler_hijack.py
@@ -359,14 +359,14 @@ class RepetitionPenaltyLogitsProcessorWithRange(LogitsProcessor):
         return scores
 
 
-def get_logits_warper_patch(self, generation_config):
+def get_logits_warper_patch(self, generation_config, **kwargs):
 
     # Parameter sanitization
     if isinstance(generation_config.temperature, int):
         generation_config.temperature = float(generation_config.temperature)  # Must be float
 
     # Get the original warpers
-    warpers = self._get_logits_warper_old(generation_config)
+    warpers = self._get_logits_warper_old(generation_config, **kwargs)
 
     # Replace temperature with our modified class.
     # Currently, it behaves identically to the original.