From e09aaedd0e74f2fc6e2a53f233914803c65e127c Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sat, 24 Dec 2022 10:25:58 +0100
Subject: Training update

---
 training/ti.py | 2 --
 1 file changed, 2 deletions(-)

(limited to 'training')

diff --git a/training/ti.py b/training/ti.py
index a5e407b..8b2fdd6 100644
--- a/training/ti.py
+++ b/training/ti.py
@@ -18,8 +18,6 @@ def patch_trainable_embeddings(text_encoder: CLIPTextModel, new_ids: list[int]):
 
     text_encoder.text_model.embeddings = text_embeddings
 
-    return text_embeddings
-
 
 class TrainableEmbeddings(CLIPTextEmbeddings):
     def __init__(self, config: CLIPTextConfig, new_ids: list[int]):
-- 
cgit v1.2.3-54-g00ecf