From 776213e99da4ec389575e797d93de8d8960fa010 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sun, 9 Apr 2023 16:21:52 +0200
Subject: Update

---
 training/strategy/ti.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'training')

diff --git a/training/strategy/ti.py b/training/strategy/ti.py
index 289d6bd..9cdc1bb 100644
--- a/training/strategy/ti.py
+++ b/training/strategy/ti.py
@@ -119,7 +119,7 @@ def textual_inversion_strategy_callbacks(
             ema_embeddings.step(text_encoder.text_model.embeddings.token_override_embedding.parameters())
 
         if use_emb_decay and w is not None:
-            lr = lrs["emb"] or lrs["0"]
+            lr = lrs["emb"] if "emb" in lrs else lrs["0"]
             lambda_ = emb_decay * lr
 
             if lambda_ != 0:
-- 
cgit v1.2.3-54-g00ecf