()
| 165 | |
| 166 | |
| 167 | def parse_args(): |
| 168 | parser = argparse.ArgumentParser('', add_help=False) |
| 169 | parser.add_argument('--paths', help='+ separated list of checkpoints', required=True, |
| 170 | type=str) |
| 171 | parser.add_argument('--save_path', help='folder name to save optimized weights', default='optimized_logs', |
| 172 | type=str) |
| 173 | parser.add_argument('--categories', help='+ separated list of categories of the models', required=True, |
| 174 | type=str) |
| 175 | parser.add_argument('--prompts', help='prompts for composition model (can be a file or string)', default=None, |
| 176 | type=str) |
| 177 | parser.add_argument('--ckpt', required=True, |
| 178 | type=str) |
| 179 | parser.add_argument('--regularization_prompt', default='./data/regularization_captions.txt', |
| 180 | type=str) |
| 181 | return parser.parse_args() |
| 182 | |
| 183 | |
| 184 | if __name__ == "__main__": |
no outgoing calls
no test coverage detected