Hi, recently I found that there might be a bug in the function convert_to_fp32().
|
return recursively_apply(_is_fp16_tensor, tensor, test_type=_is_fp16_tensor) |
I think the correct one should be
return recursively_apply(_convert_to_fp32, tensor, test_type=_is_fp16_tensor)
Thanks.
Hi, recently I found that there might be a bug in the function convert_to_fp32().
accelerate/src/accelerate/utils.py
Line 258 in 1b1463f
I think the correct one should be
return recursively_apply(_convert_to_fp32, tensor, test_type=_is_fp16_tensor)Thanks.