From 32b93da875e95b8033fe3c493eaaa7bbc9a14048 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Wed, 29 Jun 2022 17:10:08 +0200 Subject: [PATCH] begin conversion script --- .../convert_ldm_to_diffusers.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py diff --git a/src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py b/src/diffusers/pipelines/latent_diffusion_uncond/convert_ldm_to_diffusers.py new file mode 100644 index 00000000..3c512fba --- /dev/null +++ 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() +