format codes

wenet-e2e · Aug 13, 2024 · 1c1300e · 1c1300e
1 parent 223d284
commit 1c1300e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/utils/train_utils.py b/wenet/utils/train_utils.py
@@ -923,4 +923,4 @@ def reinit_lora(model, args, configs, tokenizer, seed=777):
             reinit_lora_modules(name, module, lora_config, **additional_kwargs)
     # lora_init_model needs to be saved, w0 = w0 - A0 * B0
     save_checkpoint(model, os.path.join(args.model_dir, "lora_init.pt"),
-                    infos={"tag":"lora_init", **configs})
+                    infos={"tag": "lora_init", **configs})