Skip to content

Commit d07df5a

Browse files
committed
Fixed 0.pmin_pdecay_nmin extraction
1 parent ae0a627 commit d07df5a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

common/speculative.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -155,9 +155,9 @@ llama_tokens common_speculative_gen_draft(
155155
const int i_start = std::max<int>(0, (int) prompt_tgt.size() - n_ctx);
156156

157157
// Extract parameters packed in p_min (format: 0.pmin_pdecay_nmin)
158-
const float p_min = floorf(params.p_min * 100) / 100; // First 2 decimal places
159-
const float p_decay = floorf(params.p_min * 10000) / 100 - p_min; // Next 2 decimal places
160-
const int n_min = roundf((params.p_min * 100000) - (p_min * 100000) - (p_decay * 1000)); // Last digit
158+
const float p_min = floorf(params_p_min * 100) / 100; // First 2 decimal places
159+
const float p_decay = floorf((params_p_min - p_min) * 10000) / 100; // Next 2 decimal places
160+
const int n_min = floorf((params_p_min - p_min - (p_decay / 100)) * 100000); // Last digit
161161

162162
printf("p_min=%f, p_decay=%f, n_min=%d\n", p_min, p_decay, n_min);
163163

0 commit comments

Comments
 (0)