diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py index 342cc32e33dd29d44d37cfd38f849c6fbd256e52..3cdb54e4834f11c13e9132e51b3c9a59bf26f64b 100644 --- a/dan/predict/prediction.py +++ b/dan/predict/prediction.py @@ -358,7 +358,6 @@ def run( } for current, next_token in pairwise(index + [0]) ] - result["confidences"]["total"] = np.around(np.mean(char_confidences), 2) for level in confidence_score_levels: