diff options
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 |