And perhaps related this issue: Smolagents Error: probability tensor contains either `inf`, `nan` or element < 0