feat: add ovis image support (#1057)

This commit is contained in:
leejet
2025-12-07 12:32:56 +08:00
committed by GitHub
parent bfbb929790
commit 2f0bd31a84
11 changed files with 230 additions and 71 deletions

View File

@@ -1056,6 +1056,9 @@ SDVersion ModelLoader::get_sd_version() {
if (tensor_storage.name.find("model.diffusion_model.double_stream_modulation_img.lin.weight") != std::string::npos) {
return VERSION_FLUX2;
}
if (tensor_storage.name.find("model.diffusion_model.double_blocks.0.img_mlp.gate_proj.weight") != std::string::npos) {
return VERSION_OVIS_IMAGE;
}
if (tensor_storage.name.find("model.diffusion_model.cap_embedder.0.weight") != std::string::npos) {
return VERSION_Z_IMAGE;
}