rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Fix for badwordIDs

Browse Source
This commit is contained in:
ebolam
2023-05-11 14:14:52 -04:00
parent 4605d10c37
commit e9c845dc2a
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modeling/inference_models/parents/hf.py
View File

@@ -17,6 +17,7 @@ class HFInferenceModel(InferenceModel):
self.model = None self.model = None
self.tokenizer = None self.tokenizer = None
self.badwordsids = koboldai_settings.badwordsids_default
def is_valid(self, model_name, model_path, menu_path): def is_valid(self, model_name, model_path, menu_path):
try: try:

2
modeling/inference_models/parents/hf_torch.py
View File

@@ -245,7 +245,7 @@ class HFTorchInferenceModel(HFInferenceModel):
len(prompt_tokens) + max_new, utils.koboldai_vars.max_length len(prompt_tokens) + max_new, utils.koboldai_vars.max_length
), ),
repetition_penalty=1.0, repetition_penalty=1.0,
bad_words_ids=utils.koboldai_vars.badwordsids bad_words_ids=self.badwordsids
+ additional_bad_words_ids, + additional_bad_words_ids,
use_cache=True, use_cache=True,
num_return_sequences=batch_count, num_return_sequences=batch_count,