| 191 | |
| 192 | |
| 193 | def parse_args() -> argparse.Namespace: |
| 194 | parser = argparse.ArgumentParser(description="Inference with VAE") |
| 195 | parser.add_argument( |
| 196 | "--input_image", |
| 197 | type=str, |
| 198 | required=True, |
| 199 | help="Path to the input image for inference.", |
| 200 | ) |
| 201 | parser.add_argument( |
| 202 | "--pretrained_model_name_or_path", |
| 203 | type=str, |
| 204 | required=True, |
| 205 | help="Path to pretrained VAE model.", |
| 206 | ) |
| 207 | parser.add_argument( |
| 208 | "--revision", |
| 209 | type=str, |
| 210 | default=None, |
| 211 | help="Model version.", |
| 212 | ) |
| 213 | parser.add_argument( |
| 214 | "--variant", |
| 215 | type=str, |
| 216 | default=None, |
| 217 | help="Model file variant, e.g., 'fp16'.", |
| 218 | ) |
| 219 | parser.add_argument( |
| 220 | "--subfolder", |
| 221 | type=str, |
| 222 | default=None, |
| 223 | help="Subfolder in the model file.", |
| 224 | ) |
| 225 | parser.add_argument( |
| 226 | "--use_cuda", |
| 227 | action="store_true", |
| 228 | help="Use CUDA if available.", |
| 229 | ) |
| 230 | parser.add_argument( |
| 231 | "--use_tiny_nn", |
| 232 | action="store_true", |
| 233 | help="Use tiny neural network.", |
| 234 | ) |
| 235 | return parser.parse_args() |
| 236 | |
| 237 | |
| 238 | # EXAMPLE USAGE: |