diff --git a/accel_sdxl_gen_img.py b/accel_sdxl_gen_img.py index 66ac7f9c..1c7c871f 100644 --- a/accel_sdxl_gen_img.py +++ b/accel_sdxl_gen_img.py @@ -2860,8 +2860,8 @@ def main(args): def setup_parser() -> argparse.ArgumentParser: - parser = argparse.ArgumentParser() - + parser = train_network.setup_parser() + sdxl_train_util.add_sdxl_training_arguments(parser) add_logging_arguments(parser) parser.add_argument("--prompt", type=str, default=None, help="prompt / プロンプト")