From f0014ab01c7a51bfa0f269c676404d48112f924b Mon Sep 17 00:00:00 2001 From: RoomWithOutRoof <166608075+Jah-yee@users.noreply.github.com> Date: Wed, 18 Mar 2026 09:03:48 +0800 Subject: [PATCH] fix: mutable default argument in LogitsBiasProcessor (#7426) --- modules/transformers_loader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/transformers_loader.py b/modules/transformers_loader.py index 63758ad7..7f521b8c 100644 --- a/modules/transformers_loader.py +++ b/modules/transformers_loader.py @@ -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]