Merge pull request #641 from kaibioinfo/patch-1

fix typo in sdxl_train_textual_inversion
This commit is contained in:
Kohya S
2023-07-20 08:28:11 +09:00
committed by GitHub

View File

@@ -103,8 +103,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