diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-05 13:26:32 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-05 13:26:32 +0100 |
| commit | 3396ca881ed3f3521617cd9024eea56975191d32 (patch) | |
| tree | 3189c3bbe77b211152d11b524d0fe3a7016441ee /train_ti.py | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-3396ca881ed3f3521617cd9024eea56975191d32.tar.gz textual-inversion-diff-3396ca881ed3f3521617cd9024eea56975191d32.tar.bz2 textual-inversion-diff-3396ca881ed3f3521617cd9024eea56975191d32.zip | |
Update
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/train_ti.py b/train_ti.py index 164cf67..98385dd 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -814,13 +814,13 @@ def main(): | |||
| 814 | 814 | ||
| 815 | loop = partial( | 815 | loop = partial( |
| 816 | run_model, | 816 | run_model, |
| 817 | vae=vae, | 817 | vae, |
| 818 | noise_scheduler=noise_scheduler, | 818 | noise_scheduler, |
| 819 | unet=unet, | 819 | unet, |
| 820 | prompt_processor=prompt_processor, | 820 | prompt_processor, |
| 821 | num_class_images=args.num_class_images, | 821 | args.num_class_images, |
| 822 | prior_loss_weight=args.prior_loss_weight, | 822 | args.prior_loss_weight, |
| 823 | seed=args.seed, | 823 | args.seed, |
| 824 | ) | 824 | ) |
| 825 | 825 | ||
| 826 | # We need to initialize the trackers we use, and also store our configuration. | 826 | # We need to initialize the trackers we use, and also store our configuration. |