0cc4m / KoboldAI

GNU Affero General Public License v3.0
150 stars 31 forks source link

Hook up use_default_badwordids in exllama #70

Closed pi6am closed 1 year ago

pi6am commented 1 year ago

Use the value of the use_default_badwordids setting to configure bad_words_ids. Also add square brackets to bad_words_ids if use_default_badwordids is True. Fix an issue with attempting to use the tokenizer too early, and fix an exception populating Lua bridge data when zero tokens are generated, which can now happen if use_default_badwordids is False and the first token generated is EOS. Eliminate the need for trim_count by moving the EOS early-out up.