diff options
| author | Volpeon <git@volpeon.ink> | 2023-05-11 18:37:43 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-05-11 18:37:43 +0200 |
| commit | f0a171923cc8240177302f3dccb6177a2c646ab3 (patch) | |
| tree | 15f11293c048c8c9b5d625dc27bb8662e15e685e /train_lora.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-f0a171923cc8240177302f3dccb6177a2c646ab3.tar.gz textual-inversion-diff-f0a171923cc8240177302f3dccb6177a2c646ab3.tar.bz2 textual-inversion-diff-f0a171923cc8240177302f3dccb6177a2c646ab3.zip | |
Update
Diffstat (limited to 'train_lora.py')
| -rw-r--r-- | train_lora.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py index 70fbae4..737af58 100644 --- a/train_lora.py +++ b/train_lora.py | |||
| @@ -46,7 +46,7 @@ torch.backends.cudnn.benchmark = True | |||
| 46 | torch._dynamo.config.log_level = logging.WARNING | 46 | torch._dynamo.config.log_level = logging.WARNING |
| 47 | 47 | ||
| 48 | hidet.torch.dynamo_config.use_tensor_core(True) | 48 | hidet.torch.dynamo_config.use_tensor_core(True) |
| 49 | # hidet.torch.dynamo_config.use_attention(True) | 49 | hidet.torch.dynamo_config.use_attention(True) |
| 50 | hidet.torch.dynamo_config.search_space(0) | 50 | hidet.torch.dynamo_config.search_space(0) |
| 51 | 51 | ||
| 52 | 52 | ||