Function get_args

scripts/convert_skyreelsv2_to_diffusers.py:571–576

()

def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--model_type", type=str, default=None)
    parser.add_argument("--output_path", type=str, required=True)
    parser.add_argument("--dtype", default="fp32")
    return parser.parse_args()


DTYPE_MAPPING = {