Skip to content

Commit 33d6ce0

Browse files
committed
seqrep: Rebase, deal with logging changes.
1 parent 38d218d commit 33d6ce0

File tree

3 files changed

+4
-7
lines changed

3 files changed

+4
-7
lines changed

common/common.cpp

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -108,12 +108,9 @@ void seqrep_sampler_params_init(llama_sampler_seqrep_params * params) {
108108
params->tolerance_half_step_cost = 1.0f;
109109
}
110110

111-
void seqrep_sampler_params_dump(FILE * fp, llama_sampler_seqrep_params * params) {
112-
if (fp == NULL) {
113-
return;
114-
}
111+
void seqrep_sampler_params_dump(const llama_sampler_seqrep_params * params) {
115112
assert(params != NULL);
116-
fprintf(fp, "seqrep(last_n = %d, min_length = %zd, start_offset = %zd, presence_penalty = %.4f, length_penalty = %.4f, tolerance = %.4f, mid_word_scale = %.4f, tolerance_match_credit = %.4f, tolerance_half_step_cost = %.4f, flags = %d)",
113+
LOG_TEE("seqrep(last_n = %d, min_length = %zd, start_offset = %zd, presence_penalty = %.4f, length_penalty = %.4f, tolerance = %.4f, mid_word_scale = %.4f, tolerance_match_credit = %.4f, tolerance_half_step_cost = %.4f, flags = %d)",
117114
params->last_n, params->min_length, params->start_offset, params->presence_penalty,
118115
params->length_penalty, params->tolerance, params->mid_word_scale, params->tolerance_match_credit,
119116
params->tolerance_half_step_cost, params->flags);

common/common.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,5 +168,5 @@ void dump_non_result_info_yaml(
168168
const std::string & timestamp, const std::vector<int> & prompt_tokens, const char * model_desc);
169169

170170
void seqrep_sampler_params_init(llama_sampler_seqrep_params * params);
171-
void seqrep_sampler_params_dump(FILE * fp, llama_sampler_seqrep_params * params);
171+
void seqrep_sampler_params_dump(const llama_sampler_seqrep_params * params);
172172
bool seqrep_sampler_params_parse(char * s, llama_sampler_seqrep_params * params);

examples/main/main.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -424,7 +424,7 @@ int main(int argc, char ** argv) {
424424
params.repeat_last_n, params.repeat_penalty, params.presence_penalty, params.frequency_penalty, params.top_k, params.tfs_z, params.top_p, params.typical_p, params.temp, params.mirostat, params.mirostat_eta, params.mirostat_tau);
425425
LOG_TEE("generate: n_ctx = %d, n_batch = %d, n_predict = %d, n_keep = %d\n", n_ctx, params.n_batch, params.n_predict, params.n_keep);
426426
for (auto & sr_params : params.seqrep_params) {
427-
seqrep_sampler_params_dump(stderr, &sr_params);
427+
seqrep_sampler_params_dump(&sr_params);
428428
}
429429
LOG_TEE("\n\n");
430430

0 commit comments

Comments
 (0)