fix vae pt script

0f04e799 · Patrick von Platen · 1051ca81 · 0f04e799
Commit 0f04e799 authored Feb 07, 2023 by Patrick von Platen
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

scripts/convert_vae_pt_to_diffusers.py scripts/convert_vae_pt_to_diffusers.py +2 -0

No files found.
--- 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)