oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-03-24 22:34:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: mutable default argument in LogitsBiasProcessor (#7426)

Browse source
This commit is contained in:
RoomWithOutRoof 2026-03-18 09:03:48 +08:00 committed by GitHub
parent 0f5053c0fb
commit f0014ab01c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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/transformers_loader.py
View file

@ -44,8 +44,8 @@ class Stream(transformers.StoppingCriteria):
class LogitsBiasProcessor(LogitsProcessor):
def __init__(self, logit_bias={}):
self.logit_bias = logit_bias
def __init__(self, logit_bias=None):
self.logit_bias = logit_bias if logit_bias is not None else {}
if self.logit_bias:
self.keys = list([int(key) for key in self.logit_bias.keys()])
values = [self.logit_bias[str(key)] for key in self.keys]