diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-13 09:40:34 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-13 09:40:34 +0100 |
| commit | b33ac00de283fe45edba689990dc96a5de93cd1e (patch) | |
| tree | a3106f2e482f9e4b2ab9d9ff49faf0b529278f50 /infer.py | |
| parent | Dreambooth: Support loading Textual Inversion embeddings (diff) | |
| download | textual-inversion-diff-b33ac00de283fe45edba689990dc96a5de93cd1e.tar.gz textual-inversion-diff-b33ac00de283fe45edba689990dc96a5de93cd1e.tar.bz2 textual-inversion-diff-b33ac00de283fe45edba689990dc96a5de93cd1e.zip | |
Add support for resume in Textual Inversion
Diffstat (limited to 'infer.py')
| -rw-r--r-- | infer.py | 2 |
1 files changed, 1 insertions, 1 deletions
| @@ -31,7 +31,7 @@ torch.backends.cudnn.benchmark = True | |||
| 31 | 31 | ||
| 32 | 32 | ||
| 33 | default_args = { | 33 | default_args = { |
| 34 | "model": None, | 34 | "model": "stabilityai/stable-diffusion-2-1", |
| 35 | "precision": "fp32", | 35 | "precision": "fp32", |
| 36 | "ti_embeddings_dir": "embeddings_ti", | 36 | "ti_embeddings_dir": "embeddings_ti", |
| 37 | "output_dir": "output/inference", | 37 | "output_dir": "output/inference", |