transformers

[flax] unfreeze initial cache in gpt models

#14535

Merged