Policy-Mask: Budget-Aware Adaptation of Diffusion Features to Discriminative Tasks