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 |