vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2025-09-10 09:04:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

bugfixes

Browse source
This commit is contained in:
Concedo 2024-03-05 19:16:54 +08:00
parent 4eb3a95cbb
commit d910f2354c
3 changed files with 14 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
otherarch/sdcpp/model.cpp
View file

@ -675,10 +675,13 @@ bool ModelLoader::init_from_file(const std::string& file_path, const std::string
} else if (is_safetensors_file(file_path)) {
LOG_INFO("load %s using safetensors format", file_path.c_str());
return init_from_safetensors_file(file_path, prefix);
} else if (is_zip_file(file_path)) {
LOG_INFO("load %s using checkpoint format", file_path.c_str());
return init_from_ckpt_file(file_path, prefix);
} else {
}
//disable ckpt loading
// else if (is_zip_file(file_path)) {
// LOG_INFO("load %s using checkpoint format", file_path.c_str());
// return init_from_ckpt_file(file_path, prefix);
// } else
{
LOG_WARN("unknown format %s", file_path.c_str());
return false;
}