Skip to content

Commit 0f04e79

Browse files
fix vae pt script
1 parent 1051ca8 commit 0f04e79

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

scripts/convert_vae_pt_to_diffusers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,4 +146,6 @@ def vae_pt_to_vae_diffuser(
146146
parser.add_argument("--vae_pt_path", default=None, type=str, required=True, help="Path to the VAE.pt to convert.")
147147
parser.add_argument("--dump_path", default=None, type=str, required=True, help="Path to the VAE.pt to convert.")
148148

149+
args = parser.parse_args()
150+
149151
vae_pt_to_vae_diffuser(args.vae_pt_path, args.dump_path)

0 commit comments

Comments
 (0)