Fixes https://github.com/huggingface/datasets/issues/6566

Let me know if there's any tests I need to clear.
Update features.py to avoid bfloat16 unsupported error
I think not all torch tensors should be converted to float, what if it's a tensor of integers for example ?
Maybe you can check for the tensor dtype before converting