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

---
 textual_inversion.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'textual_inversion.py')

diff --git a/textual_inversion.py b/textual_inversion.py
index 7ac9638..d6be522 100644
--- a/textual_inversion.py
+++ b/textual_inversion.py
@@ -31,6 +31,7 @@ logger = get_logger(__name__)
 
 
 torch.backends.cuda.matmul.allow_tf32 = True
+torch.backends.cudnn.benchmark = True
 
 
 def parse_args():
-- 
cgit v1.2.3-54-g00ecf