waifu-collective
/
toolbox
2
1
Fork 0
Code Issues 2 Pull Requests Releases Wiki Activity

chore: update module list in build_dataset.py

This commit is contained in:
11b 2022-12-23 16:37:58 -03:00
parent a16673ebe0
commit d91367e902
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
waifu/scripts/build_dataset.py
View File

@ -15,8 +15,11 @@ from waifu.modules import BaseModule
# metaprogramming trickery to build this list out instead. # metaprogramming trickery to build this list out instead.
DEFAULT_MODULE_LIST = [ DEFAULT_MODULE_LIST = [
"characterai_pdm:CharacterAiPDM", "characterai_pdm:CharacterAiPDM",
"kajiwoto_pdm:KajiwotoPDM", "discord_vdm:DiscordVDM",
# "kajiwoto_vdm:KajiwotoVDM", # KajiwotoPDM has a bunch of garbage I need to filter, disabling in favor
# of the vanilla dialogue module for now.
# "kajiwoto_pdm:KajiwotoPDM",
"kajiwoto_vdm:KajiwotoVDM",
"light_dialogue_pdm:LightDialoguePDM", "light_dialogue_pdm:LightDialoguePDM",
] ]
DEFAULT_MODULES_STRING = ",".join(DEFAULT_MODULE_LIST) DEFAULT_MODULES_STRING = ",".join(DEFAULT_MODULE_LIST)
@ -72,7 +75,8 @@ def main() -> None:
# Print a newline to visually separate different episodes. # Print a newline to visually separate different episodes.
if idx != 1: if idx != 1:
print() print()
print("--- new episode ---") print("---| New Episode |---")
print("---------------------")
print(episode) print(episode)
sys.exit() sys.exit()