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

fix typo in sdxl_train_textual_inversion

Browse Source 
bug appears when continue training on an existing TI
This commit is contained in:
kaibioinfo
2023-07-12 15:06:20 +02:00
committed by GitHub
parent 8fa5fb2816
commit a7ce2633f3
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

@@ -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