feat: add moss-image support (#2148)

This commit is contained in:
leejet authored and GitHub committed 2026-10-17 03:38:38 +08:00
1 parent af0fc72c96
commit 68285df319
11 files changed
+508 -10

No files matched your search

+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;