def get_model_name()

in Example-Project/src/tune_gpt.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

def get_model_name(model_size):
    if 'large' in model_size:
        model_name = '774M'

    elif 'medium' in model_size: 
        model_name = "355M"

    elif 'small' in model_size:
        model_name = "124M"
    
    return model_name