Fix to enter BestSampler route when temperature is 0. (#1659)

OpenNMT · Apr 10, 2024 · 1deef09 · 1deef09
1 parent 7d63eea
commit 1deef09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/decoding.cc b/src/decoding.cc
@@ -1007,7 +1007,7 @@ namespace ctranslate2 {
 
   static std::unique_ptr<const Sampler>
   make_sampler(const DecodingOptions& options) {
-    if (options.sampling_topk == 1)
+    if (options.sampling_topk == 1 || options.sampling_temperature == 0.0)
       return std::make_unique<BestSampler>();
     else
       return std::make_unique<RandomSampler>(options.sampling_topk,