From 5c115a212e40ff177c734351601f9babe29419ce Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 5 Jan 2023 22:05:25 +0100
Subject: Added EMA to TI

---
 train_dreambooth.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'train_dreambooth.py')

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 4d1e0a3..c355ea1 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -638,7 +638,7 @@ def main():
     if args.train_text_encoder:
         print(f"Training entire text encoder.")
 
-        embeddings.make_permanent()
+        embeddings.persist()
         text_encoder.text_model.embeddings.temp_token_embedding.requires_grad_(False)
     else:
         print(f"Training added text embeddings")
-- 
cgit v1.2.3-54-g00ecf