mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 22:35:09 +00:00
feat: added function to load training config with .toml
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user