Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-21 15:48:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

Small fix to make transformers 4.42 functional

Browse Source
This commit is contained in:
oobabooga 2024-06-27 17:05:29 -07:00
parent 66090758df
commit 9dbcb1aeea
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sampler_hijack.py
View File

@ -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.