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 /textual_inversion.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 'textual_inversion.py')
| -rw-r--r-- | textual_inversion.py | 1 |
1 files changed, 1 insertions, 0 deletions
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__) | |||
| 31 | 31 | ||
| 32 | 32 | ||
| 33 | torch.backends.cuda.matmul.allow_tf32 = True | 33 | torch.backends.cuda.matmul.allow_tf32 = True |
| 34 | torch.backends.cudnn.benchmark = True | ||
| 34 | 35 | ||
| 35 | 36 | ||
| 36 | def parse_args(): | 37 | def parse_args(): |