def freeze_layers()

in scripts/ft_gemma3n_image_vt.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def freeze_layers(model):
    for name, param in model.named_parameters():
        if "attn" in name:
            param.requires_grad = True
        else:
            param.requires_grad = False
    return model