From 1b46c661320bd3c488dd697c866c45b71a7df884 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 28 Aug 2023 07:17:21 +0000 Subject: [PATCH] make style --- examples/textual_inversion/textual_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/textual_inversion/textual_inversion.py b/examples/textual_inversion/textual_inversion.py index d6d815d951..ea3bc1f9eb 100644 --- a/examples/textual_inversion/textual_inversion.py +++ b/examples/textual_inversion/textual_inversion.py @@ -889,7 +889,7 @@ def main(): if global_step % args.save_steps == 0: weight_name = ( 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" ) save_path = os.path.join(args.output_dir, weight_name)