transformers

fix: jamba cache fails to use torch.nn.module

#32894

Merged