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-70-g09d2