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

missed a thing

Browse source
This commit is contained in:
Concedo 2023-04-25 23:02:08 +08:00
parent 235daf4016
commit 8c9c218609
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
gpttype_adapter.cpp
View file

@ -412,7 +412,7 @@ generation_outputs gpttype_generate(const generation_inputs inputs, generation_o
{
n_vocab = gpt2_ctx_v2.hparams.n_vocab;
}
else if(file_format == FileFormat::NEOX_1)
else if(file_format == FileFormat::NEOX_1 || file_format == FileFormat::NEOX_2)
{
n_vocab = neox_ctx.hparams.n_vocab;
}