AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix name when saving text inversion embeddings in dreambooth advanced scripts (#8927)

Browse Source 
update
This commit is contained in:
Dhruv Nair
2024-07-23 19:51:20 +05:30
committed by GitHub
parent cf55dcf0ff
commit 3bb1fd6fc0
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py
View File

@@ -1302,7 +1302,7 @@ def main(args):
text_encoder_lora_layers=text_encoder_one_lora_layers_to_save,
)
if args.train_text_encoder_ti:
embedding_handler.save_embeddings(f"{output_dir}/{args.output_dir}_emb.safetensors")
embedding_handler.save_embeddings(f"{args.output_dir}/{Path(args.output_dir).name}_emb.safetensors")
def load_model_hook(models, input_dir):
unet_ = None

2
examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
View File

@@ -1627,7 +1627,7 @@ def main(args):
text_encoder_2_lora_layers=text_encoder_two_lora_layers_to_save,
)
if args.train_text_encoder_ti:
embedding_handler.save_embeddings(f"{output_dir}/{args.output_dir}_emb.safetensors")
embedding_handler.save_embeddings(f"{args.output_dir}/{Path(args.output_dir).name}_emb.safetensors")
def load_model_hook(models, input_dir):
unet_ = None