def gpu_size(wildcards, attempt):
    """Return model size in gb based on {prot} and {model}"""

    prot = wildcards.get('prot', 'V3')
    model = wildcards.get('model', 'protbert')

    if 'bert' not in model:
        return 0
    if prot == 'V3':
        return 9*attempt
    return 12*attempt