| 51 | |
| 52 | |
| 53 | def parse_args(): |
| 54 | parser = argparse.ArgumentParser(description="Simple example of a training script.") |
| 55 | parser.add_argument( |
| 56 | "--pretrained_model_name_or_path", |
| 57 | type=str, |
| 58 | default=None, |
| 59 | help="Path to pretrained model or model identifier from huggingface.co/models.", |
| 60 | ) |
| 61 | parser.add_argument( |
| 62 | "--tokenizer_name", |
| 63 | type=str, |
| 64 | default=None, |
| 65 | help="Pretrained tokenizer name or path if not the same as model_name", |
| 66 | ) |
| 67 | parser.add_argument( |
| 68 | "--train_data_dir", |
| 69 | type=str, |
| 70 | default=None, |
| 71 | help="A folder containing the training data.", |
| 72 | ) |
| 73 | parser.add_argument( |
| 74 | "--placeholder_token", |
| 75 | type=str, |
| 76 | default=None, |
| 77 | help="A token to use as a placeholder for the concept.", |
| 78 | ) |
| 79 | parser.add_argument( |
| 80 | "--initializer_token", |
| 81 | type=str, |
| 82 | default=None, |
| 83 | help="A token to use as initializer word.", |
| 84 | ) |
| 85 | parser.add_argument( |
| 86 | "--learnable_property", |
| 87 | type=str, |
| 88 | default="object", |
| 89 | help="Choose between 'object' and 'style'", |
| 90 | ) |
| 91 | parser.add_argument( |
| 92 | "--repeats", |
| 93 | type=int, |
| 94 | default=100, |
| 95 | help="How many times to repeat the training data.", |
| 96 | ) |
| 97 | parser.add_argument( |
| 98 | "--output_dir", |
| 99 | type=str, |
| 100 | default="text-inversion-model", |
| 101 | help="The output directory where the model predictions and checkpoints will be written.", |
| 102 | ) |
| 103 | parser.add_argument( |
| 104 | "--seed", type=int, default=None, help="A seed for reproducible training." |
| 105 | ) |
| 106 | parser.add_argument( |
| 107 | "--resolution", |
| 108 | type=int, |
| 109 | default=512, |
| 110 | help=( |