diff options
| author | Volpeon <git@volpeon.ink> | 2022-10-22 11:05:12 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-10-22 11:05:12 +0200 |
| commit | 0995b6d9b16c3b0ac4971e0d2ef4cf8f3ee050e8 (patch) | |
| tree | ce50fd284a73d9f62041fc997c0db4d7ea5f04be /textual_inversion.py | |
| parent | Add optional TI functionality to Dreambooth (diff) | |
| download | textual-inversion-diff-0995b6d9b16c3b0ac4971e0d2ef4cf8f3ee050e8.tar.gz textual-inversion-diff-0995b6d9b16c3b0ac4971e0d2ef4cf8f3ee050e8.tar.bz2 textual-inversion-diff-0995b6d9b16c3b0ac4971e0d2ef4cf8f3ee050e8.zip | |
Training update
Diffstat (limited to 'textual_inversion.py')
| -rw-r--r-- | textual_inversion.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/textual_inversion.py b/textual_inversion.py index fe56d36..c42762f 100644 --- a/textual_inversion.py +++ b/textual_inversion.py | |||
| @@ -520,7 +520,6 @@ def main(): | |||
| 520 | prompt_processor = PromptProcessor(tokenizer, text_encoder) | 520 | prompt_processor = PromptProcessor(tokenizer, text_encoder) |
| 521 | 521 | ||
| 522 | if args.gradient_checkpointing: | 522 | if args.gradient_checkpointing: |
| 523 | unet.enable_gradient_checkpointing() | ||
| 524 | text_encoder.gradient_checkpointing_enable() | 523 | text_encoder.gradient_checkpointing_enable() |
| 525 | 524 | ||
| 526 | # slice_size = unet.config.attention_head_dim // 2 | 525 | # slice_size = unet.config.attention_head_dim // 2 |