VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 14:34:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix to work train_network with fine-tuning dataset

Browse Source
This commit is contained in:
Kohya S
2023-06-28 07:50:53 +09:00
parent 07d5c71090
commit 71a6d49d06
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
library/config_util.py
View File

@@ -262,11 +262,11 @@ class ConfigSanitizer:
{"subsets": [self.cn_subset_schema]},
)
if support_dreambooth and support_finetuning and support_controlnet:
if support_dreambooth and support_finetuning:
def validate_flex_dataset(dataset_config: dict):
subsets_config = dataset_config.get("subsets", [])
if all(["conditioning_data_dir" in subset for subset in subsets_config]):
if support_controlnet and all(["conditioning_data_dir" in subset for subset in subsets_config]):
return Schema(self.cn_dataset_schema)(dataset_config)
# check dataset meets FT style
# NOTE: all FT subsets should have "metadata_file"
@@ -347,6 +347,7 @@ class BlueprintGenerator:
# runtime_params is for parameters which is only configurable on runtime, such as tokenizer
def generate(self, user_config: dict, argparse_namespace: argparse.Namespace, **runtime_params) -> Blueprint:
print(user_config)
sanitized_user_config = self.sanitizer.sanitize_user_config(user_config)
sanitized_argparse_namespace = self.sanitizer.sanitize_argparse_namespace(argparse_namespace)