diff options
| author | Volpeon <git@volpeon.ink> | 2022-09-28 15:05:30 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-09-28 15:05:30 +0200 |
| commit | 5b54788842cdd7b342bd60d6944158009130b4d4 (patch) | |
| tree | ebeb9fbb2c2e4d4d2e406d3ad86b01deb2b1525a /textual_inversion.py | |
| parent | Infer script: Store args, better path/file names (diff) | |
| download | textual-inversion-diff-5b54788842cdd7b342bd60d6944158009130b4d4.tar.gz textual-inversion-diff-5b54788842cdd7b342bd60d6944158009130b4d4.tar.bz2 textual-inversion-diff-5b54788842cdd7b342bd60d6944158009130b4d4.zip | |
Improved sample output and progress bars
Diffstat (limited to 'textual_inversion.py')
| -rw-r--r-- | textual_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/textual_inversion.py b/textual_inversion.py index 81f1cf5..399d876 100644 --- a/textual_inversion.py +++ b/textual_inversion.py | |||
| @@ -581,7 +581,7 @@ def main(): | |||
| 581 | 581 | ||
| 582 | # TODO (patil-suraj): laod scheduler using args | 582 | # TODO (patil-suraj): laod scheduler using args |
| 583 | noise_scheduler = DDPMScheduler( | 583 | noise_scheduler = DDPMScheduler( |
| 584 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000, tensor_format="pt" | 584 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000 |
| 585 | ) | 585 | ) |
| 586 | 586 | ||
| 587 | datamodule = CSVDataModule( | 587 | datamodule = CSVDataModule( |