提交 9646f1aa 编写于 作者: M Megvii Engine Team

fix(lite/load_and_run): fix invalid model type caused by oss model magic change

GitOrigin-RevId: 1d071f999e70111fb6f63a5b8f333ccd9b498161
上级 623477fd
......@@ -24,12 +24,12 @@ ModelType ModelBase::get_model_type(std::string model_path) {
fclose(fin);
// get model type
// uint32_t MGB_MAGIC = 0x5342474D
std::string tag(buf);
ModelType type;
if (tag.substr(0, 7) == std::string("mgb0001") ||
tag.substr(0, 8) == std::string("mgb0000a") ||
tag.substr(0, 4) == std::string("MGBS") ||
tag.substr(0, 4) == std::string("MGBC") ||
tag.substr(0, 8) == std::string("mgbtest0")) {
type = ModelType::MEGDL_MODEL;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册