()
| 124 | import os |
| 125 | |
| 126 | def parse_args(): |
| 127 | parser = argparse.ArgumentParser(description="Simple example of a training script.") |
| 128 | |
| 129 | parser.add_argument( |
| 130 | "--real_data_dir", |
| 131 | type=str, |
| 132 | default=None, |
| 133 | required=True, |
| 134 | ) |
| 135 | |
| 136 | parser.add_argument( |
| 137 | "--prompts_path", |
| 138 | type=str, |
| 139 | default=None, |
| 140 | required=True, |
| 141 | ) |
| 142 | parser.add_argument( |
| 143 | "--fake_data_dir", |
| 144 | type=str, |
| 145 | default=None, |
| 146 | required=True, |
| 147 | ) |
| 148 | |
| 149 | parser.add_argument( |
| 150 | "--image_align_only_training_prompt", |
| 151 | default=False, |
| 152 | action="store_true", |
| 153 | # required=True, |
| 154 | ) |
| 155 | |
| 156 | parser.add_argument( |
| 157 | "--text_align_wo_training_prompt", |
| 158 | default=False, |
| 159 | action="store_true", |
| 160 | # required=True, |
| 161 | ) |
| 162 | |
| 163 | |
| 164 | args = parser.parse_args() |
| 165 | return args |
| 166 | |
| 167 | args = parse_args() |
| 168 |