| 56 | |
| 57 | |
| 58 | def parse_args(): |
| 59 | parser = argparse.ArgumentParser('', add_help=False) |
| 60 | parser.add_argument('--ckpt', help='target string for query', |
| 61 | type=str) |
| 62 | parser.add_argument('--delta_ckpt', help='target string for query', default=None, |
| 63 | type=str) |
| 64 | parser.add_argument('--from-file', help='path to prompt file', default='./', |
| 65 | type=str) |
| 66 | parser.add_argument('--prompt', help='prompt to generate', default=None, |
| 67 | type=str) |
| 68 | parser.add_argument("--compress", action='store_true') |
| 69 | parser.add_argument("--sdxl", action='store_true') |
| 70 | parser.add_argument("--batch_size", default=5, type=int) |
| 71 | parser.add_argument('--freeze_model', help='crossattn or crossattn_kv', default='crossattn_kv', |
| 72 | type=str) |
| 73 | return parser.parse_args() |
| 74 | |
| 75 | |
| 76 | if __name__ == "__main__": |