Update common/sampling.cpp

Co-authored-by: compilade <[email protected]>
jart · May 21, 2024 · aa3094c · aa3094c
1 parent 6b17898
commit aa3094c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/common/sampling.cpp b/common/sampling.cpp
@@ -195,9 +195,7 @@ static llama_token llama_sampling_sample_impl(
     llama_token id = 0;
     // Get a pointer to the logits
     float * logits = llama_get_logits_ith(ctx_main, idx);
-    if (!logits) {
-        throw std::runtime_error("llama_get_logits_ith failed");
-    }
+    GGML_ASSERT(logits); // already checked in llama_sampling_prepare 
 
     if (temp < 0.0) {
         // greedy sampling, with probs