def parse_args()

in training/generate_images.py [0:0]

• 40 lines of code
• 1 McCabe index (conditional complexity)

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--pretrained_model_name_or_path",
        type=str,
        default=None,
        required=True,
        help="Path to pretrained model or model identifier from huggingface.co/models.",
    )
    parser.add_argument(
        "--revision",
        type=str,
        default=None,
        required=False,
        help="Revision of pretrained model identifier from huggingface.co/models.",
    )
    parser.add_argument(
        "--variant",
        type=str,
        default=None,
        help="Variant of the model files of the pretrained model identifier from huggingface.co/models, 'e.g.' fp16",
    )
    parser.add_argument("--style_descriptor", type=str, default="[V]")
    parser.add_argument(
        "--load_transformer_from",
        type=str,
        required=False,
        default=None,
    )
    parser.add_argument(
        "--load_transformer_lora_from",
        type=str,
        required=False,
        default=None,
    )
    parser.add_argument("--device", type=str, default='cuda')
    parser.add_argument("--batch_size", type=int, default=1)
    parser.add_argument("--write_images_to", type=str, required=True)
    args = parser.parse_args()
    return args