Merge remote-tracking branch 'upstream/united' into 4bit-plugin

This commit is contained in:
0cc4m
2023-07-23 07:18:58 +02:00
8 changed files with 44 additions and 25 deletions

View File

@@ -330,6 +330,11 @@ class HFInferenceModel(InferenceModel):
if any(c in str(k) for c in "[]")
]
try:
self.badwordsids.remove([self.tokenizer.pad_token_id])
except:
pass
if utils.koboldai_vars.newlinemode == "n":
self.badwordsids.append([self.tokenizer.eos_token_id])