Skip to content

Commit

Permalink
Fix for koboldcpp kobldai model backend. Generate function changed in…
Browse files Browse the repository at this point in the history
…puts. Passthrough now passes the proper variables. (#641)
  • Loading branch information
ebolam authored Jan 27, 2024
1 parent ed09a85 commit af8503e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions class.py
Original file line number Diff line number Diff line change
Expand Up @@ -301,10 +301,10 @@ def _raw_generate(
# Store context in memory to use it for comparison with generated content
utils.koboldai_vars.lastctx = decoded_prompt

self.input_queue.put({'command': 'generate', 'data': [(decoded_prompt,"",max_new,utils.koboldai_vars.max_length,
gen_settings.temp,int(gen_settings.top_k),gen_settings.top_a,gen_settings.top_p,
gen_settings.typical,gen_settings.tfs,gen_settings.rep_pen,gen_settings.rep_pen_range),
{"sampler_order": gen_settings.sampler_order, "use_default_badwordsids": utils.koboldai_vars.use_default_badwordsids}
self.input_queue.put({'command': 'generate', 'data': [(decoded_prompt,), {'max_length': max_new, 'max_context_length': utils.koboldai_vars.max_length,
'temperature': gen_settings.temp, 'top_k': int(gen_settings.top_k), 'top_a': gen_settings.top_a, 'top_p': gen_settings.top_p,
'typical_p': gen_settings.typical, 'tfs': gen_settings.tfs, 'rep_pen': gen_settings.rep_pen, 'rep_pen_range': gen_settings.rep_pen_range,
"sampler_order": gen_settings.sampler_order, "use_default_badwordsids": utils.koboldai_vars.use_default_badwordsids}
]})

#genresult = koboldcpp.generate(decoded_prompt,"",max_new,utils.koboldai_vars.max_length,
Expand Down

0 comments on commit af8503e

Please sign in to comment.