fix `'int' object is not callable` warning

2023-01-25 11:06:52 +01:00 · 2023-01-25 11:06:52 +01:00 · 448848d5c2
parent 09347779be
commit 448848d5c2
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -483,7 +483,7 @@ def main(args):
        tokenizer = CLIPTokenizer.from_pretrained(model_root_folder, subfolder="tokenizer", use_fast=False)
    except Exception as e:
        traceback.print_exc()
-        logging.ERROR(" * Failed to load checkpoint *")
+        logging.error(" * Failed to load checkpoint *")

    if args.gradient_checkpointing:
        unet.enable_gradient_checkpointing()