Merge pull request #31 from damian0815/patch-1

fix `'int' object is not callable` warning
2023-01-25 09:29:31 -08:00 · 2023-01-25 09:29:31 -08:00 · b4847d6f45
parent 09347779be 448848d5c2
commit b4847d6f45
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()