feat: add moss-image support (#2148)

This commit is contained in:
leejet
2026-10-17 03:38:38 +08:00
committed by GitHub
parent af0fc72c96
commit 68285df319
11 changed files with 508 additions and 10 deletions
+5
View File
@@ -444,6 +444,11 @@ SDVersion ModelLoader::get_sd_version() const {
return VERSION_FLUX3;
}
if (tensor_storage_map.count("model.diffusion_model.video_embedder.weight") &&
tensor_storage_map.count("model.diffusion_model.shared_block_modulation.weight")) {
return VERSION_MOSS_IMAGE;
}
bool has_multiple_encoders = false;
bool is_unet = false;