Skip to content

GitLab

Commit 1b46c661 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 03135898
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/textual_inversion/textual_inversion.py
View file @ 1b46c661
...@@ -889,7 +889,7 @@ def main(): ...@@ -889,7 +889,7 @@ def main():
if global_step % args.save_steps == 0: if global_step % args.save_steps == 0:
weight_name = ( weight_name = (
f"learned_embeds-steps-{global_step}.bin" f"learned_embeds-steps-{global_step}.bin"
if args.no_safe_serialization if args.no_safe_serialization
else f"learned_embeds-steps-{global_step}.safetensors" else f"learned_embeds-steps-{global_step}.safetensors"
) )
save_path = os.path.join(args.output_dir, weight_name) save_path = os.path.join(args.output_dir, weight_name)
......
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