From c5cc1318c2a7597fe62d3379e50187d0b0f22538 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 1 Dec 2022 13:45:21 +0100
Subject: Update

---
 infer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'infer.py')

diff --git a/infer.py b/infer.py
index ab5f247..eabeb5e 100644
--- a/infer.py
+++ b/infer.py
@@ -16,6 +16,7 @@ from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion
 
 
 torch.backends.cuda.matmul.allow_tf32 = True
+torch.backends.cudnn.benchmark = True
 
 
 default_args = {
@@ -37,7 +38,7 @@ default_cmds = {
     "height": 512,
     "batch_size": 1,
     "batch_num": 1,
-    "steps": 50,
+    "steps": 30,
     "guidance_scale": 7.0,
     "seed": None,
     "config": None,
-- 
cgit v1.2.3-54-g00ecf