Skip to content

Commit 3bb1fd6

Browse files
authored
Fix name when saving text inversion embeddings in dreambooth advanced scripts (huggingface#8927)
update
1 parent cf55dcf commit 3bb1fd6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1302,7 +1302,7 @@ def save_model_hook(models, weights, output_dir):
13021302
text_encoder_lora_layers=text_encoder_one_lora_layers_to_save,
13031303
)
13041304
if args.train_text_encoder_ti:
1305-
embedding_handler.save_embeddings(f"{output_dir}/{args.output_dir}_emb.safetensors")
1305+
embedding_handler.save_embeddings(f"{args.output_dir}/{Path(args.output_dir).name}_emb.safetensors")
13061306

13071307
def load_model_hook(models, input_dir):
13081308
unet_ = None

examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1627,7 +1627,7 @@ def save_model_hook(models, weights, output_dir):
16271627
text_encoder_2_lora_layers=text_encoder_two_lora_layers_to_save,
16281628
)
16291629
if args.train_text_encoder_ti:
1630-
embedding_handler.save_embeddings(f"{output_dir}/{args.output_dir}_emb.safetensors")
1630+
embedding_handler.save_embeddings(f"{args.output_dir}/{Path(args.output_dir).name}_emb.safetensors")
16311631

16321632
def load_model_hook(models, input_dir):
16331633
unet_ = None

0 commit comments

Comments
 (0)