begin conversion script
This commit is contained in:
parent
597b7ae2fb
commit
32b93da875
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue