676 Commits

Author SHA1 Message Date
Gnome Ann
8c63f7c034 Change |nil in Lua type annotations to ? 2021-12-21 23:43:56 -05:00
Gnome Ann
380b54167a Make transformers warpers dynamically update their parameters
So that if you change, e.g., `top_p`, from a Lua generation modifier or
from the settings menu during generation, the rest of the generation
will use the new setting value instead of retaining the settings it had
when generation began.
2021-12-21 22:12:24 -05:00
Gnome Ann
91b6289897 Upload Lua pattern matching libraries 2021-12-21 19:35:57 -05:00
henk717
41d7c2acfe
Merge pull request #48 from VE-FORBRYDERNE/patch
Disable `low_cpu_mem_usage` when using GPT-2
2021-12-21 02:45:44 +01:00
Gnome Ann
caef3b7460 Disable low_cpu_mem_usage when using GPT-2
Attempting to use transformers 4.11.0's experimental `low_cpu_mem_usage`
feature with GPT-2 models usually results in the output repeating a
token over and over or otherwise containing an incoherent response.
2021-12-20 19:54:19 -05:00
henk717
7b56940ed7
Merge pull request #47 from VE-FORBRYDERNE/scripting
Lua API fixes
2021-12-20 04:32:25 +01:00
henk717
8ee250e422
Merge pull request #46 from VE-FORBRYDERNE/wi-patch
Fix a bug where WI entries sometimes can't be deleted if the current story was loaded from a save
2021-12-20 04:31:32 +01:00
Gnome Ann
7dd319491d Fix compute_context() method of world info folders 2021-12-19 20:46:37 -05:00
Gnome Ann
341b153360 Lua API fixes
* `print()` and `warn()` now work correctly with `nil` arguments
* Typo: `gpt-neo-1.3M` has been corrected to `gpt-neo-1.3B`
* Regeneration is no longer triggered when writing to `keysecondary` of
  a non-selective key
* Handle `genamt` changes in generation modifier properly
* Writing to `kobold.settings.numseqs` from a generation modifier no
  longer affects
* Formatting options in `kobold.settings` have been fixed
* Added aliases for setting names
* Fix behaviour of editing story chunks from a generation modifier
* Warnings are now yellow instead of red
* kobold.logits is now the raw logits prior to being filtered, like
  the documentation says, rather than after being filtered
* Some erroneous comments and error messages have been corrected
* These parts of the API have now been implemented properly:
    * `compute_context()` methods
    * `kobold.authorsnote`
    * `kobold.restart_generation()`
2021-12-19 20:18:28 -05:00
Gnome Ann
6aba869fb7 Make sure uninitialized WI entries are given UIDs when loading saves 2021-12-18 18:00:06 -05:00
henk717
4bb5e59d82
Merge pull request #45 from VE-FORBRYDERNE/scripting
Fix behaviour of `kobold.outputs` with read-only and no prompt gen
2021-12-17 19:45:31 +01:00
Gnome Ann
12718dbe24 Try long-polling first, then try websocket
This makes it so that SocketIO uses long polling to set up the
connection before switching to websocket, instead of immediately using
websocket.

This seems to resolve issues where the browser sometimes can't connect
to the websocket server until the window has been open for a minute.
2021-12-17 13:18:47 -05:00
Gnome Ann
769333738d Fix behaviour of kobold.outputs with read-only and no prompt gen 2021-12-17 12:59:01 -05:00
henk717
d15b43e20e Not always a list of strings
kobold.outputs must be a 1D list of strings, but sometimes its still blank. In those cases rather than throwing an error and crashing the scripting its better if it does nothing.
2021-12-16 12:54:10 +01:00
henk717
6d9063fb8b No Prompt Gen
Allow people to enter a prompt without generating anything by the AI. Combined with the always add prompt this is a very useful feature that allows people to write world information first, and then do a specific action. This mimics the behavior previously seen in AI Dungeon forks where it prompts for world information and then asks an action and can be particularly useful for people who want the prompt to always be part of the generation.
2021-12-16 12:47:44 +01:00
henk717
f3b4ecabca
Merge pull request #44 from VE-FORBRYDERNE/patch
Fix an error that occurs when all layers are on second GPU
2021-12-16 01:43:03 +01:00
henk717
e3d9c2d690 New download machanism
Automatically converts Huggingface cache models to full models on (down)load.
WARNING: Does wipe old cache/ dir inside the KoboldAI folder, make a backup before you run these models if you are bandwith constraint.
2021-12-16 01:40:04 +01:00
Gnome Ann
19d2356253 Fix an error that occurs when all layers are on second GPU 2021-12-15 19:03:49 -05:00
henk717
5e3e3f3578 Fix float16 models
Forcefully convert float16 models to work on the CPU
2021-12-16 00:31:51 +01:00
henk717
46b0473229
Merge pull request #43 from VE-FORBRYDERNE/dynamic-scan-patch
Dynamic scan patch
2021-12-15 09:45:07 +01:00
Gnome Ann
9097aac4a8 Show full stack trace for generator errors to help in diagnosing errors 2021-12-15 02:03:08 -05:00
Gnome Ann
2687135e05 Fix a strange bug where max tokens was capped at 1024
This seems to be related to the model config files, because only certain
models have this problem, and replacing ALL configuration files of a
"bad" model with those of a "good" model of the same type would fix the
problem.

Shouldn't be required anymore.
2021-12-15 00:45:41 -05:00
Gnome Ann
1551c45ba4 Prevent dynamic scanning from generating too many tokens 2021-12-14 23:39:04 -05:00
Gnome Ann
629988ce13 Fix a problem with the Lua regeneration API
It was an egregious typo that caused tokens to be rearranged on
regeneration.
2021-12-14 23:04:03 -05:00
henk717
56679d775f Update update-kobold.bat
Improved reliability
2021-12-14 18:53:52 +01:00
henk717
6670168a47 Update aiserver.py 2021-12-14 16:26:23 +01:00
henk717
cb98462b02 Replace update with switch
My idea to checkout the used branch failed making the updater obsolete, rebranded the switcher to the updater.
2021-12-14 16:00:35 +01:00
henk717
c701fdce1d Update update-kobold.bat 2021-12-14 15:55:44 +01:00
henk717
2d1561aa55 Version Switcher
Allows people to easily switch between different versions of KoboldAI. The stable one, United or their own. Compatible with my earlier update script.
2021-12-14 15:24:56 +01:00
henk717
b824ce0b33 Update Script
Updates KoboldAI to the latest official version, if you want to use United you first manually need to be on that git branch otherwise it gets overwritten with the old one.
2021-12-14 14:26:04 +01:00
henk717
c5dec67f13 Fix Netbase on Colab
Apparently Colab does not properly have netbase which we now use for the proper websocket support, now the installer forces it to be correctly installed so we don't crash on launch.
2021-12-14 04:02:33 +01:00
henk717
0f06cee272 Don't upload developer userscripts
We may want to bundle these at some point, but in that case you should make an exception like we do for the sample story. The same applies to Softprompts.
2021-12-14 02:55:16 +01:00
henk717
c5ade0333a Userscript support on GDrive
Make the Colab's create and map a userscripts folder.
2021-12-14 02:51:39 +01:00
henk717
18ddd77337 Apply VE's changes to colab 2021-12-14 02:13:43 +01:00
henk717
a0ccbda6b1
Merge pull request #42 from VE-FORBRYDERNE/scripting
Lua scripting
2021-12-14 02:12:09 +01:00
Gnome Ann
6e6e0b2b4d Allow Lua to stop generation from input modifier 2021-12-13 19:32:01 -05:00
Gnome Ann
e9ed8602b2 Add a "corescript" setting 2021-12-13 19:28:33 -05:00
Gnome Ann
e5bb20cc8f Fix Lua regeneration system 2021-12-13 19:17:18 -05:00
Gnome Ann
462040ed6f Restore missing loadsettings() call 2021-12-13 18:39:33 -05:00
Gnome Ann
661cca63e8 Make sure stopping criteria still work with dynamic scan off 2021-12-13 18:10:51 -05:00
Gnome Ann
338d437ea3 Use eventlet instead of gevent-websocket 2021-12-13 17:19:04 -05:00
Gnome Ann
fb6762bc1a Add "AVAILABLE" and "ACTIVE" headings to userscript menu 2021-12-13 12:45:52 -05:00
Gnome Ann
ed9c2a4d52 Fix a bug that occurs when userscript doesn't have all 3 modifiers 2021-12-13 11:50:10 -05:00
Gnome Ann
34c52a1a23 Remove escape characters from all error messages 2021-12-13 11:47:34 -05:00
Gnome Ann
11f9866dbe Enable more of the IO library in Lua sandbox
Also changes the Lua warning color to red.
2021-12-13 11:22:58 -05:00
Gnome Ann
28e86563b8 Change self.scores to scores in aiserver.py 2021-12-13 11:18:01 -05:00
Gnome Ann
a06cc3d37b Make sure the "Lua error" GUI message is always visible 2021-12-13 03:21:08 -05:00
Gnome Ann
d870f193b2 Create userscripts folder if it doesn't exist 2021-12-13 02:33:36 -05:00
Gnome Ann
82e149ee02 Catch Lua errors properly 2021-12-13 02:32:09 -05:00
Gnome Ann
5f06d20085 Format Lua printed messages and warnings 2021-12-13 01:59:53 -05:00