Skip to content

Commit

Permalink
Merge pull request kohya-ss#641 from kaibioinfo/patch-1
Browse files Browse the repository at this point in the history
fix typo in sdxl_train_textual_inversion
  • Loading branch information
kohya-ss authored Jul 19, 2023
2 parents 225e871 a7ce263 commit 771f33d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions sdxl_train_textual_inversion.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,8 103,8 @@ def load_weights(self, file):
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
Expand Down

0 comments on commit 771f33d

Please sign in to comment.