Skip to content

GitLab

Unverified Commit 3bb1fd6f authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

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

update
parent cf55dcf0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py
View file @ 3bb1fd6f
...@@ -1302,7 +1302,7 @@ def main(args): ...@@ -1302,7 +1302,7 @@ def main(args):
text_encoder_lora_layers=text_encoder_one_lora_layers_to_save, text_encoder_lora_layers=text_encoder_one_lora_layers_to_save,
) )
if args.train_text_encoder_ti: 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): def load_model_hook(models, input_dir):
unet_ = None unet_ = None
......
examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
View file @ 3bb1fd6f
...@@ -1627,7 +1627,7 @@ def main(args): ...@@ -1627,7 +1627,7 @@ def main(args):
text_encoder_2_lora_layers=text_encoder_two_lora_layers_to_save, text_encoder_2_lora_layers=text_encoder_two_lora_layers_to_save,
) )
if args.train_text_encoder_ti: 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): def load_model_hook(models, input_dir):
unet_ = None unet_ = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment