[timesfm2_5] fix timesfm2.5 loss (#44331)
* fix timesfm2.5 loss
* Update src/transformers/models/timesfm2_5/modular_timesfm2_5.py
Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>
* fix typo
* use input_ff_layer.weight.dtype to cast
* more dtype
* fix tests
* formatting
* address review
* Update src/transformers/models/timesfm2_5/modular_timesfm2_5.py
Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>
* fix tests
* formatting
---------
Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>
