implement new training master script

2026-09-20 01:31:13 +02:00 · 2023-03-19 12:13:16 -04:00
parent a856ace900
commit 53db51194f
21 changed files with 1194 additions and 28 deletions
@@ -190,7 +190,6 @@ if __name__ == '__main__':
    options.unet_lr = args.unetlr
    options.text_encoder_lr = args.textlr
    options.train_batch_size = args.batch
-    options.network_alpha = args.alpha
    log.info({ 'train lora args': vars(options) })
    transformers.logging.set_verbosity_error()
    mem_stats()