qwen2 warning FA

This commit is contained in:
Concedo 2024-07-09 20:53:25 +08:00
parent 116d5fe58e
commit 0dd3907940
3 changed files with 7 additions and 14 deletions

View file

@ -310,9 +310,9 @@ void print_tok_vec(std::vector<float> &embd)
{
fileformatmeta->model_architecture = GGUFArch::ARCH_SOLAR;
}
else if(modelarch=="llama" && freq_base_train==10000.0f)
else if(modelarch=="qwen2")
{
fileformatmeta->model_architecture = GGUFArch::ARCH_MISTRAL_LLAMA_1_AND_2;
fileformatmeta->model_architecture = GGUFArch::ARCH_QWEN2;
}
printf("Arch Category: %d\n",fileformatmeta->model_architecture);