[Dúvida] Erro 'float' object has no attribute 'size'

Olá, Bom dia.

Estou tendo o seguinte erro ao rodar o comando:

from accelerate import notebook_launcher

notebook_launcher(funcao_treinamento)

assim que termina a primeira sessão de treinamento (50%) ele me retorna esse erro:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-45-72dda5bcf0c7> in <cell line: 3>()
      1 from accelerate import notebook_launcher
      2 
----> 3 notebook_launcher(funcao_treinamento)

4 frames
~/.cache/huggingface/modules/evaluate_modules/metrics/evaluate-metric--f1/0ca73f6cf92ef5a268320c697f7b940d1030f8471714bffdb6856c641b818974/f1.py in _compute(self, predictions, references, labels, pos_label, average, sample_weight)
    128             references, predictions, labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight
    129         )
--> 130         return {"f1": float(score) if score.size == 1 else score}

AttributeError: 'float' object has no attribute 'size'

tentei pelo notebook do professor e também acontece o mesmo erro, estou deixando passar algo?