AttributeError: 'ReusableKVCache' object has no attribute 'head_dim'

[uogbuji]

This is due to an upstream change in mlx_lm. While investgating & fixing I also caught & applied based on this earlier patch.

Fixed in main. Will release Toolio 0.3.1 soon today.

cc @choccccy @olajiide