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.
DEFAULT_MODULE_LIST = [
"characterai_pdm:CharacterAiPDM",
"kajiwoto_pdm:KajiwotoPDM",
# "kajiwoto_vdm:KajiwotoVDM",
"discord_vdm:DiscordVDM",
# 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",
]
DEFAULT_MODULES_STRING = ",".join(DEFAULT_MODULE_LIST)
@ -72,7 +75,8 @@ def main() -> None:
# Print a newline to visually separate different episodes.
if idx != 1:
print()
print("--- new episode ---")
print("---| New Episode |---")
print("---------------------")
print(episode)
sys.exit()