From 0995b6d9b16c3b0ac4971e0d2ef4cf8f3ee050e8 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sat, 22 Oct 2022 11:05:12 +0200
Subject: Training update

---
 textual_inversion.py | 1 -
 1 file changed, 1 deletion(-)

(limited to 'textual_inversion.py')

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():
     prompt_processor = PromptProcessor(tokenizer, text_encoder)
 
     if args.gradient_checkpointing:
-        unet.enable_gradient_checkpointing()
         text_encoder.gradient_checkpointing_enable()
 
     # slice_size = unet.config.attention_head_dim // 2
-- 
cgit v1.2.3-54-g00ecf