diff options
author | Volpeon <git@volpeon.ink> | 2022-12-01 13:45:21 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2022-12-01 13:45:21 +0100 |
commit | c5cc1318c2a7597fe62d3379e50187d0b0f22538 (patch) | |
tree | 66f2b939c5498849692836e368fce481e7eaf3d2 /infer.py | |
parent | Update (diff) | |
download | textual-inversion-diff-c5cc1318c2a7597fe62d3379e50187d0b0f22538.tar.gz textual-inversion-diff-c5cc1318c2a7597fe62d3379e50187d0b0f22538.tar.bz2 textual-inversion-diff-c5cc1318c2a7597fe62d3379e50187d0b0f22538.zip |
Update
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 3 |
1 files changed, 2 insertions, 1 deletions
@@ -16,6 +16,7 @@ from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion | |||
16 | 16 | ||
17 | 17 | ||
18 | torch.backends.cuda.matmul.allow_tf32 = True | 18 | torch.backends.cuda.matmul.allow_tf32 = True |
19 | torch.backends.cudnn.benchmark = True | ||
19 | 20 | ||
20 | 21 | ||
21 | default_args = { | 22 | default_args = { |
@@ -37,7 +38,7 @@ default_cmds = { | |||
37 | "height": 512, | 38 | "height": 512, |
38 | "batch_size": 1, | 39 | "batch_size": 1, |
39 | "batch_num": 1, | 40 | "batch_num": 1, |
40 | "steps": 50, | 41 | "steps": 30, |
41 | "guidance_scale": 7.0, | 42 | "guidance_scale": 7.0, |
42 | "seed": None, | 43 | "seed": None, |
43 | "config": None, | 44 | "config": None, |