()
| 55 | |
| 56 | |
| 57 | def parse_args(): |
| 58 | parser = argparse.ArgumentParser('', add_help=False) |
| 59 | parser.add_argument('--delta_ckpt', help='path of checkpoint to compress', |
| 60 | type=str) |
| 61 | parser.add_argument('--ckpt', help='path of pretrained model checkpoint', |
| 62 | type=str) |
| 63 | parser.add_argument("--diffuser", action='store_true') |
| 64 | return parser.parse_args() |
| 65 | |
| 66 | |
| 67 | if __name__ == "__main__": |