toolbox/waifu/datasets
TearGosling ea162de2e0 feat: add SODA dataset
* Very first prototype of SODA dataset support

I'm also bringing over the version of PromptConstants from the dev branch due to needing CHAT_START_TOKEN

* More flexibility when fetching speaker names

* Make SODA a PDM instead of a VDM

* Swap order of speakers based on relation

* Oh, and fix a typo too

* Bugfix
2023-01-08 15:48:52 -03:00
..
__init__.py chore: enable strict mode on mypy, add isort check 2022-12-17 21:31:41 -03:00
characterai.py fix: human/bot messages being incorrectly labeled as eachother 2022-12-24 17:58:33 -03:00
kajiwoto.py chore: fix linter/style problems 2023-01-01 11:50:23 -03:00
light_dialogue.py feat: add the LIGHT dataset and VDM 2022-12-18 17:29:15 -03:00
soda.py feat: add SODA dataset 2023-01-08 15:48:52 -03:00