Update whisper.cc (#1846)

Added logits to WhisperGenerationResult
OpenNMT · Feb 10, 2025 · 6a3dc63 · 6a3dc63
1 parent 8dee2a2
commit 6a3dc63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)))
             + ")";
         })