feat: added function to load training config with .toml

This commit is contained in:
Linaqruf
2023-03-12 11:52:37 +07:00
parent 7c1cf7f4ea
commit 44d4cfb453
7 changed files with 96 additions and 4 deletions

View File

@@ -1598,6 +1598,8 @@ def add_training_arguments(parser: argparse.ArgumentParser, support_dreambooth:
'dpmsolver++', 'dpmsingle',
'k_lms', 'k_euler', 'k_euler_a', 'k_dpm_2', 'k_dpm_2_a'],
help=f'sampler (scheduler) type for sample images / サンプル出力時のサンプラー(スケジューラ)の種類')
parser.add_argument("--config_file", type=str, default=None, help="using .toml instead of args to pass hyperparameter")
if support_dreambooth:
# DreamBooth training