Ensure to write use_jp_extra option to config.json

This commit is contained in:
litagin02
2024-03-12 19:10:43 +09:00
parent 456fe00ae4
commit 4acfc2977e
3 changed files with 6 additions and 3 deletions

View File

@@ -87,6 +87,9 @@ def initialize(
config["train"]["bf16_run"] = False # デフォルトでFalseのはずだが念のため
# 今はデフォルトであるが、以前は非JP-Extra版になくバグの原因になるので念のため
config["data"]["use_jp_extra"] = use_jp_extra
model_path = os.path.join(dataset_path, "models")
if os.path.exists(model_path):
logger.warning(