mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
begin conversion script
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user