From 0f04e799dc915fbbcb1ccd0e8fe5ce30633f0214 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 7 Feb 2023 08:34:19 +0000 Subject: [PATCH] fix vae pt script --- scripts/convert_vae_pt_to_diffusers.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/convert_vae_pt_to_diffusers.py b/scripts/convert_vae_pt_to_diffusers.py index e82ad85001..ca79c8a0c6 100644 --- a/scripts/convert_vae_pt_to_diffusers.py +++ b/scripts/convert_vae_pt_to_diffusers.py @@ -146,4 +146,6 @@ if __name__ == "__main__": parser.add_argument("--vae_pt_path", default=None, type=str, required=True, help="Path to the VAE.pt to convert.") parser.add_argument("--dump_path", default=None, type=str, required=True, help="Path to the VAE.pt to convert.") + args = parser.parse_args() + vae_pt_to_vae_diffuser(args.vae_pt_path, args.dump_path)