diff options
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(): |