The noun capability is normally followed by the preposition 'for'. Example:
The capability for humor will take you a long way in this job.
As a runner, he is fortunate to have a capability for endurance.
My capability for mechanical things is sorely lacking; I'm more of an intellectual genius.
Note: The adjective form, capable, is normally followed by the preposition 'of'. Example:
I need a sewing machine capable of embroidery stitches.