fixed bbs -1 and allow bbs = 2048

This commit is contained in:
Concedo 2023-08-06 17:47:05 +08:00
parent 18bb0ab127
commit bcfdd0e662
3 changed files with 11 additions and 5 deletions

View file

@ -347,6 +347,10 @@ ModelLoadResult gpttype_load_model(const load_model_inputs inputs, FileFormat in
debugmode = inputs.debugmode;
unbanTokens = inputs.unban_tokens;
blasbatchsize = inputs.blasbatchsize;
if(blasbatchsize<=0)
{
blasbatchsize = 8;
}
params.memory_f16 = inputs.f16_kv;
params.n_ctx = inputs.max_context_length;