diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-13 07:14:24 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-13 07:14:24 +0200 |
| commit | a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e (patch) | |
| tree | 6a695b2b5a73cebc35ff9e581c70f1a0e75b62e8 /training/strategy/ti.py | |
| parent | Experimental convnext discriminator support (diff) | |
| download | textual-inversion-diff-a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e.tar.gz textual-inversion-diff-a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e.tar.bz2 textual-inversion-diff-a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e.zip | |
Update
Diffstat (limited to 'training/strategy/ti.py')
| -rw-r--r-- | training/strategy/ti.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/training/strategy/ti.py b/training/strategy/ti.py index 08af89d..ca7cc3d 100644 --- a/training/strategy/ti.py +++ b/training/strategy/ti.py | |||
| @@ -158,8 +158,7 @@ def textual_inversion_strategy_callbacks( | |||
| 158 | unet_.to(dtype=orig_unet_dtype) | 158 | unet_.to(dtype=orig_unet_dtype) |
| 159 | text_encoder_.to(dtype=orig_text_encoder_dtype) | 159 | text_encoder_.to(dtype=orig_text_encoder_dtype) |
| 160 | 160 | ||
| 161 | del unet_ | 161 | del unet_, text_encoder_ |
| 162 | del text_encoder_ | ||
| 163 | 162 | ||
| 164 | if torch.cuda.is_available(): | 163 | if torch.cuda.is_available(): |
| 165 | torch.cuda.empty_cache() | 164 | torch.cuda.empty_cache() |