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_dreambooth.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_dreambooth.py')
| -rw-r--r-- | train_dreambooth.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index 1fd86b1..4d1e0a3 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
| @@ -861,13 +861,13 @@ def main(): | |||
| 861 | 861 | ||
| 862 | loop = partial( | 862 | loop = partial( |
| 863 | run_model, | 863 | run_model, |
| 864 | vae=vae, | 864 | vae, |
| 865 | noise_scheduler=noise_scheduler, | 865 | noise_scheduler, |
| 866 | unet=unet, | 866 | unet, |
| 867 | prompt_processor=prompt_processor, | 867 | prompt_processor, |
| 868 | num_class_images=args.num_class_images, | 868 | args.num_class_images, |
| 869 | prior_loss_weight=args.prior_loss_weight, | 869 | args.prior_loss_weight, |
| 870 | seed=args.seed, | 870 | args.seed, |
| 871 | ) | 871 | ) |
| 872 | 872 | ||
| 873 | # We need to initialize the trackers we use, and also store our configuration. | 873 | # We need to initialize the trackers we use, and also store our configuration. |