Skip to content

GitLab

Commit 0f04e799 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

fix vae pt script

parent 1051ca81
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
scripts/convert_vae_pt_to_diffusers.py
View file @ 0f04e799
...@@ -146,4 +146,6 @@ if __name__ == "__main__": ...@@ -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("--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.") 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) vae_pt_to_vae_diffuser(args.vae_pt_path, args.dump_path)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment