diff options
| author | Volpeon <git@volpeon.ink> | 2022-11-03 17:56:08 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-11-03 17:56:08 +0100 |
| commit | 036e33fbde6bad7c48bb6f6b3d695b7908535c64 (patch) | |
| tree | 6b0df3deed03ccc8722763b51a9d7c2019be8e10 /infer.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-036e33fbde6bad7c48bb6f6b3d695b7908535c64.tar.gz textual-inversion-diff-036e33fbde6bad7c48bb6f6b3d695b7908535c64.tar.bz2 textual-inversion-diff-036e33fbde6bad7c48bb6f6b3d695b7908535c64.zip | |
Update
Diffstat (limited to 'infer.py')
| -rw-r--r-- | infer.py | 2 |
1 files changed, 1 insertions, 1 deletions
| @@ -234,7 +234,7 @@ def create_pipeline(model, scheduler, ti_embeddings_dir, dtype): | |||
| 234 | tokenizer=tokenizer, | 234 | tokenizer=tokenizer, |
| 235 | scheduler=scheduler, | 235 | scheduler=scheduler, |
| 236 | ) | 236 | ) |
| 237 | pipeline.aesthetic_gradient_iters = 20 | 237 | pipeline.enable_xformers_memory_efficient_attention() |
| 238 | pipeline.to("cuda") | 238 | pipeline.to("cuda") |
| 239 | 239 | ||
| 240 | print("Pipeline loaded.") | 240 | print("Pipeline loaded.") |