Quando você percebe que os NaNs vêm daquela parte do código onde você converte os valores para fp32 para evitar problemas, mas as coisas pioram muito quando você converte de volta para fp16.
Carregando detalhes do thread
Buscando os tweets originais no X para montar uma leitura limpa.
Isso normalmente leva apenas alguns segundos.