diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-22 08:30:55 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-22 08:30:55 +0100 |
| commit | c11f30e1c1a5f27504011b746c67cdc9853c8d12 (patch) | |
| tree | 954452ae1649cbb9163bdebd30beb57d68b6ca00 /training | |
| parent | Log DAdam/DAdan d (diff) | |
| download | textual-inversion-diff-c11f30e1c1a5f27504011b746c67cdc9853c8d12.tar.gz textual-inversion-diff-c11f30e1c1a5f27504011b746c67cdc9853c8d12.tar.bz2 textual-inversion-diff-c11f30e1c1a5f27504011b746c67cdc9853c8d12.zip | |
Fix
Diffstat (limited to 'training')
| -rw-r--r-- | training/functional.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/functional.py b/training/functional.py index 77f056e..ebb48ab 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -429,7 +429,7 @@ def train_loop( | |||
| 429 | try: | 429 | try: |
| 430 | import dadaptation | 430 | import dadaptation |
| 431 | 431 | ||
| 432 | isDadaptation = isinstance(optimizer, (dadaptation.DAdaptAdam, dadaptation.DAdaptAdan)) | 432 | isDadaptation = isinstance(optimizer.optimizer, (dadaptation.DAdaptAdam, dadaptation.DAdaptAdan)) |
| 433 | except ImportError: | 433 | except ImportError: |
| 434 | pass | 434 | pass |
| 435 | 435 | ||