From a7ce2633f3a9eb9488714ceade0e7009874cef9a Mon Sep 17 00:00:00 2001 From: kaibioinfo Date: Wed, 12 Jul 2023 15:06:20 +0200 Subject: [PATCH] fix typo in sdxl_train_textual_inversion bug appears when continue training on an existing TI --- sdxl_train_textual_inversion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdxl_train_textual_inversion.py b/sdxl_train_textual_inversion.py index 9df37092..942d9e74 100644 --- a/sdxl_train_textual_inversion.py +++ b/sdxl_train_textual_inversion.py @@ -115,8 +115,8 @@ class SdxlTextualInversionTrainer(train_textual_inversion.TextualInversionTraine else: data = torch.load(file, map_location="cpu") - emb_l = data.get("clib_l", None) # ViT-L text encoder 1 - emb_g = data.get("clib_g", None) # BiG-G text encoder 2 + emb_l = data.get("clip_l", None) # ViT-L text encoder 1 + emb_g = data.get("clip_g", None) # BiG-G text encoder 2 assert ( emb_l is not None or emb_g is not None