vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

erase constant seed, add changing alpha and beta parameters from command line

Browse source
This commit is contained in:
Bartosz Podkanowicz 2023-11-09 15:37:53 +01:00
parent 38c5b7ee5f
commit db2a5beef1
1 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

18
examples/contrastive/contrastive.cpp
View file

@ -11,7 +11,7 @@ int main(int argc, char ** argv) {
gpt_params params_expert;
gpt_params params_amateur;
if (argc == 1 || argv[1][0] == '-') {
printf("usage: %s EXPERT_MODEL_PATH AMATEUR_MODEL_PATH [PROMPT]\n" , argv[0]);
printf("usage: %s EXPERT_MODEL_PATH AMATEUR_MODEL_PATH [PROMPT] [alpha] [beta]\n" , argv[0]);
return 1;
}
@ -28,6 +28,17 @@ int main(int argc, char ** argv) {
params_amateur.prompt = argv[3];
}
float alpha = 0.1;
float beta = 0.5;
if(argc >= 5){
alpha = std::stof(argv[4]);
}
if(argc >= 6){
beta = std::stof(argv[5]);
}
if (params_expert.prompt.empty()) {
params_expert.prompt = "Hello my name is";
params_amateur.prompt = "Hello my name is";
@ -64,7 +75,6 @@ int main(int argc, char ** argv) {
llama_context_params ctx_params = llama_context_default_params();
ctx_params.seed = 1234;
ctx_params.n_ctx = 2048;
ctx_params.n_threads = params_expert.n_threads;
ctx_params.n_threads_batch = params_expert.n_threads_batch == -1 ? params_expert.n_threads : params_expert.n_threads_batch;
@ -138,10 +148,6 @@ int main(int argc, char ** argv) {
int n_decode = 0;
const auto t_main_start = ggml_time_us();
float alpha = 0.1;
float beta = 0.5;
while (n_cur <= n_len) {
// sample the next token
{