When resuming from a checkpoint we were printing the wrong variable o…

…n the maskgit training and so it was not properly displayed on the console. This was only a visual bug and the right variable was used for resuming.
ShoukanLabs · Sep 16, 2023 · 32d1acb · 32d1acb
1 parent 9e9a86c
commit 32d1acb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_muse_maskgit.py b/train_muse_maskgit.py
@@ -613,7 +613,7 @@ def main():
                 else:
                     ema_vae = None
 
-                print(f"Resuming VAE from latest checkpoint: {args.resume_path}")
+                print(f"Resuming VAE from latest checkpoint: {args.vae_path}")
             else:
                 accelerator.print("Resuming VAE from: ", args.vae_path)
                 ema_vae = None