Commit 32b93da8 authored by patil-suraj's avatar patil-suraj
Browse files

begin conversion script

parent 597b7ae2
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()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment