AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

fix vae pt script

Browse Source
This commit is contained in:
Patrick von Platen
2023-02-07 08:34:19 +00:00
parent 1051ca81a6
commit 0f04e799dc
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/convert_vae_pt_to_diffusers.py
View File

@@ -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)