begin conversion script

2022-06-29 17:10:08 +02:00 · 2022-06-29 17:10:08 +02:00 · 32b93da875
parent 597b7ae2fb
commit 32b93da875
1 changed files with 13 additions and 0 deletions
--- a/src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py
+++ b/src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py
@ -0,0 +1,13 @@
+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()
+