Commit graph

80 commits

Author SHA1 Message Date
oobabooga c90310e40e Small simplification 2023-01-19 00:41:57 -03:00
oobabooga 99536ef5bf Add no-stream option 2023-01-18 23:56:42 -03:00
oobabooga 116299b3ad Manual eos_token implementation 2023-01-18 22:57:39 -03:00
oobabooga 3cb30bed0a Add a "stop" button 2023-01-18 22:44:47 -03:00
oobabooga 8f27d33034 Fix another bug 2023-01-18 22:08:23 -03:00
oobabooga 6c7f187586 Minor change 2023-01-18 21:59:23 -03:00
oobabooga b3cba0b330 Bug 2023-01-18 21:54:44 -03:00
oobabooga df2e910421 Stop generating in chat mode when \nYou: is generated 2023-01-18 21:51:18 -03:00
oobabooga 022960a087 This is the correct way of sampling 1 token at a time 2023-01-18 21:37:21 -03:00
oobabooga 0f01a3b1fa Implement text streaming (#10)
Still experimental. There might be bugs.
2023-01-18 19:06:50 -03:00
oobabooga ca13acdfa0 Ensure that the chat prompt will always contain < 2048 tokens
This way, we can keep the context string at the top of the prompt
even if you keep talking to the bot for hours.

Before this commit, the prompt would be simply truncated and the
context string would eventually be lost.
2023-01-17 20:16:23 -03:00
oobabooga 6456777b09 Clean things up 2023-01-16 16:35:45 -03:00
oobabooga 3a99b2b030 Change a truncation parameter 2023-01-16 13:53:30 -03:00
oobabooga 54bf55372b Truncate prompts to 2048 characters 2023-01-16 13:43:23 -03:00
oobabooga c7a2818665
Grammar 2023-01-16 10:10:09 -03:00
oobabooga d973897021
Typo 2023-01-16 01:52:28 -03:00
oobabooga 47a20638de Don't need this 2023-01-15 23:15:30 -03:00
oobabooga b55486fa00 Reorganize things 2023-01-15 23:01:51 -03:00
oobabooga ebf4d5f506 Add --max-gpu-memory parameter for #7 2023-01-15 22:33:35 -03:00
oobabooga bb1a172da0 Fix a bug in cai mode chat 2023-01-15 19:41:25 -03:00
oobabooga e6691bd920 Make chat mode more like cai 2023-01-15 18:16:46 -03:00
oobabooga e04ecd4bce Minor improvements 2023-01-15 16:43:31 -03:00
oobabooga 027c3dd27d Allow jpg profile images 2023-01-15 15:45:25 -03:00
oobabooga afe9f77f96 Reorder parameters 2023-01-15 15:30:39 -03:00
oobabooga 88d67427e1 Implement default settings customization using a json file 2023-01-15 15:23:41 -03:00
oobabooga 6136da419c Add --cai-chat option that mimics Character.AI's interface 2023-01-15 12:20:04 -03:00
oobabooga 13b04c1b94 Add "remove last message" button to chat 2023-01-15 03:19:09 -03:00
oobabooga fd220f827f Remove annoying warnings 2023-01-15 00:39:51 -03:00
oobabooga d962e69496 Improve chat preprocessing 2023-01-14 23:50:34 -03:00
oobabooga 9a7f187b5a Improve pygmalion line breaks 2023-01-14 23:26:14 -03:00
oobabooga ecb2cc2194 Pygmalion: add checkbox for choosing whether to stop at newline or not 2023-01-13 15:02:17 -03:00
oobabooga 3a00cb1bbd Reorganize GUI elements 2023-01-13 14:28:53 -03:00
oobabooga 3f1e70d2c8 Remove the temperature slider
It was not being used by most presets.
2023-01-13 14:00:43 -03:00
oobabooga 7f93012a89 Add default names/context for pygmalion 2023-01-13 10:12:47 -03:00
oobabooga 9410486bd8 Enable the API
Let's goooooooooooooo
2023-01-11 16:43:13 -03:00
oobabooga 66f73c1b32 Remove default text from output box 2023-01-11 01:36:11 -03:00
oobabooga 01ac065d7e Implement Continue button 2023-01-11 01:33:57 -03:00
oobabooga 4b09e7e355 Sort models alphabetically 2023-01-11 01:17:20 -03:00
oobabooga d5e01c80e3 Add nice HTML output for all models 2023-01-11 01:10:11 -03:00
oobabooga b2a2ddcb15 Remove T5 support (it sucks) 2023-01-10 23:39:50 -03:00
oobabooga a236b24d24 Add --auto-devices and --load-in-8bit options for #4 2023-01-10 23:16:33 -03:00
oobabooga 3aefcfd963 Grammar 2023-01-09 19:07:47 -03:00
oobabooga 6c178b1c91 Add --listen parameter 2023-01-09 19:05:36 -03:00
oobabooga 13836a37c8 Remove unused parameter 2023-01-09 17:23:43 -03:00
oobabooga f0013ac8e9 Don't need that 2023-01-09 16:30:14 -03:00
oobabooga 00a12889e9 Refactor model loading function 2023-01-09 16:28:04 -03:00
oobabooga 980f8112a7 Small bug fix 2023-01-09 12:56:54 -03:00
oobabooga a751d7e693 Don't require GPT-J to be installed to load gpt4chan 2023-01-09 11:39:13 -03:00
oobabooga 6cbfe19c23 Submit with Shift+Enter 2023-01-09 11:22:12 -03:00
oobabooga 0e67ccf607 Implement CPU mode 2023-01-09 10:58:46 -03:00