Fix for a Bug in BitFit for T5

[prateeky2806]

Hi, I was trying to run BitFit for T5 and was getting some errors while initializing the bias parameters. It seems like in you register the bias parameter later to the linear class and initialize it first which leads to an error. The error happened here, moving L185 to the place of L179 fixes the bug.

Thanks,
Prateek