JaswaniSaniya

> if hidden_states.isinf().sum() > 0.: > inf_idx = torch.where(hidden_states.isinf()) > print("Warning: 'hidden_states' have 'inf' values at", inf_idx) > hidden_states[inf_idx[0], inf_idx[1], inf_idx[2]] = 0. This doesnt work. Do we have to...