VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-10 15:00:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #641 from kaibioinfo/patch-1

Browse Source 
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
parent 225e871819 a7ce2633f3
commit 771f33d17d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sdxl_train_textual_inversion.py
View File

@@ -103,8 +103,8 @@ class SdxlTextualInversionTrainer(train_textual_inversion.TextualInversionTraine
else: else:
data = torch.load(file, map_location="cpu") data = torch.load(file, map_location="cpu")
emb_l = data.get("clib_l", None) # ViT-L text encoder 1 emb_l = data.get("clip_l", None) # ViT-L text encoder 1
emb_g = data.get("clib_g", None) # BiG-G text encoder 2 emb_g = data.get("clip_g", None) # BiG-G text encoder 2
assert ( assert (
emb_l is not None or emb_g is not None emb_l is not None or emb_g is not None