()
| 25 | |
| 26 | |
| 27 | def get_args(): |
| 28 | parser = argparse.ArgumentParser() |
| 29 | parser.add_argument("--ckpt_path", type=str, required=True) |
| 30 | parser.add_argument("--output_path", type=str, required=True) |
| 31 | parser.add_argument("--use_motion_mid_block", action="store_true") |
| 32 | parser.add_argument("--motion_max_seq_length", type=int, default=32) |
| 33 | parser.add_argument("--save_fp16", action="store_true") |
| 34 | |
| 35 | return parser.parse_args() |
| 36 | |
| 37 | |
| 38 | if __name__ == "__main__": |
no outgoing calls
no test coverage detected