diff options
author | Volpeon <git@volpeon.ink> | 2023-04-07 09:09:46 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-04-07 09:09:46 +0200 |
commit | d952d467d31786f4a85cc4cb009934cd4ebbba71 (patch) | |
tree | 68c3f597a86ef3b98734d80cc783aa1f42fe1a41 /train_ti.py | |
parent | Update (diff) | |
download | textual-inversion-diff-d952d467d31786f4a85cc4cb009934cd4ebbba71.tar.gz textual-inversion-diff-d952d467d31786f4a85cc4cb009934cd4ebbba71.tar.bz2 textual-inversion-diff-d952d467d31786f4a85cc4cb009934cd4ebbba71.zip |
Update
Diffstat (limited to 'train_ti.py')
-rw-r--r-- | train_ti.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/train_ti.py b/train_ti.py index fc0d68c..344b412 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -750,6 +750,7 @@ def main(): | |||
750 | accelerator=accelerator, | 750 | accelerator=accelerator, |
751 | unet=unet, | 751 | unet=unet, |
752 | text_encoder=text_encoder, | 752 | text_encoder=text_encoder, |
753 | tokenizer=tokenizer, | ||
753 | vae=vae, | 754 | vae=vae, |
754 | noise_scheduler=noise_scheduler, | 755 | noise_scheduler=noise_scheduler, |
755 | dtype=weight_dtype, | 756 | dtype=weight_dtype, |
@@ -764,8 +765,6 @@ def main(): | |||
764 | global_step_offset=global_step_offset, | 765 | global_step_offset=global_step_offset, |
765 | offset_noise_strength=args.offset_noise_strength, | 766 | offset_noise_strength=args.offset_noise_strength, |
766 | # -- | 767 | # -- |
767 | tokenizer=tokenizer, | ||
768 | sample_scheduler=sample_scheduler, | ||
769 | checkpoint_output_dir=checkpoint_output_dir, | 768 | checkpoint_output_dir=checkpoint_output_dir, |
770 | use_emb_decay=args.use_emb_decay, | 769 | use_emb_decay=args.use_emb_decay, |
771 | emb_decay_target=args.emb_decay_target, | 770 | emb_decay_target=args.emb_decay_target, |
@@ -774,6 +773,7 @@ def main(): | |||
774 | ema_inv_gamma=args.ema_inv_gamma, | 773 | ema_inv_gamma=args.ema_inv_gamma, |
775 | ema_power=args.ema_power, | 774 | ema_power=args.ema_power, |
776 | ema_max_decay=args.ema_max_decay, | 775 | ema_max_decay=args.ema_max_decay, |
776 | sample_scheduler=sample_scheduler, | ||
777 | sample_batch_size=args.sample_batch_size, | 777 | sample_batch_size=args.sample_batch_size, |
778 | sample_num_batches=args.sample_batches, | 778 | sample_num_batches=args.sample_batches, |
779 | sample_num_steps=args.sample_steps, | 779 | sample_num_steps=args.sample_steps, |
@@ -863,9 +863,9 @@ def main(): | |||
863 | optimizer=optimizer, | 863 | optimizer=optimizer, |
864 | lr_scheduler=lr_scheduler, | 864 | lr_scheduler=lr_scheduler, |
865 | num_train_epochs=num_train_epochs, | 865 | num_train_epochs=num_train_epochs, |
866 | sample_frequency=sample_frequency, | ||
867 | # -- | 866 | # -- |
868 | sample_output_dir=sample_output_dir, | 867 | sample_output_dir=sample_output_dir, |
868 | sample_frequency=sample_frequency, | ||
869 | placeholder_tokens=placeholder_tokens, | 869 | placeholder_tokens=placeholder_tokens, |
870 | placeholder_token_ids=placeholder_token_ids, | 870 | placeholder_token_ids=placeholder_token_ids, |
871 | ) | 871 | ) |