fix broken sd1.x PTI caused by SDXL PTI fix

feffy380 · Apr 2, 2024 · aba39c0 · aba39c0
1 parent ef4f5ea
commit aba39c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_network.py b/train_network.py
@@ -907,7 +907,7 @@ def save_model(ckpt_name, unwrapped_nw, steps, epoch_no, embeddings_map, force_s
                         state_dict[key2] = embs[1]
                     else:
                         key = f"bundle_emb.{emb_name}.string_to_param.*"
-                        state_dict[key] = embeddings_map[emb_name]
+                        state_dict[key] = embeddings_map[emb_name][0]
 
                 if metadata_to_save is not None and len(metadata_to_save) == 0:
                     metadata_to_save = None