@@ -146,4 +146,6 @@ def vae_pt_to_vae_diffuser(
     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)