begin conversion script

32b93da8 · patil-suraj · 597b7ae2 · 32b93da8
Commit 32b93da8 authored Jun 29, 2022 by patil-suraj
Hide whitespace changes
Inline Side-by-side

Showing with 13 additions and 0 deletions

src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py ...lines/latent_diffusion_uncond/convert_ldm_to_diffusers.py +13 -0

No files found.
--- a/src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py
+++ b/src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py
+import argparse
+import torch
+from diffusers import UNetLDMModel, VQModel
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--checkpoint_path", type=str, required=True)
+    parser.add_argument("--config_path", type=str, required=True)
+    parser.add_argument("--output_path", type=str, required=True)
+    args = parser.parse_args()