From b3850259ce1e2426b84950cffc4e0c0dec3d6380 Mon Sep 17 00:00:00 2001
From: Samantha Grey <49354894+sammygrey@users.noreply.github.com>
Date: Thu, 16 Jan 2025 01:21:16 -0500
Subject: [PATCH] Update whisper.cc

Added logits to WhisperGenerationResult
---
 python/cpp/whisper.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/cpp/whisper.cc b/python/cpp/whisper.cc
index 75c9c6255..34500745c 100644
--- a/python/cpp/whisper.cc
+++ b/python/cpp/whisper.cc
@@ -133,6 +133,7 @@ namespace ctranslate2 {
           return "WhisperGenerationResult(sequences=" + std::string(py::repr(py::cast(result.sequences)))
             + ", sequences_ids=" + std::string(py::repr(py::cast(result.sequences_ids)))
             + ", scores=" + std::string(py::repr(py::cast(result.scores)))
+            + ", logits=" + std::string(py::repr(py::cast(result.logits)))
             + ", no_speech_prob=" + std::string(py::repr(py::cast(result.no_speech_prob)))
             + ")";
         })