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]