diff options
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, |