"git@developer.sourcefind.cn:OpenDAS/ktransformers.git" did not exist on "1dcb8dae5b94e4af56d023cce359e52657d0d14e"
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
...@@ -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
......
...@@ -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