cuda, metal : fix nans in soft_max (#5574)
* cuda : fix nans in soft_max * metal : fix nans in soft_max --------- Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
S
slaren committed
3a9cb4ca6408c29423373dd6cd7aa78a58286c00
Parent: 769a716
Committed by GitHub <noreply@github.com>
on 2/19/2024, 8:04:45 AM