feat: add FLUX.3 Image Klein and Klein Base support (#2143)

This commit is contained in:
leejet
2026-10-16 02:42:07 +08:00
committed by GitHub
parent e47426c044
commit 8cd731e7b4
15 changed files with 413 additions and 6 deletions
+5
View File
@@ -438,6 +438,11 @@ bool ModelLoader::init_from_diffusers_file(const std::string& file_path, const s
SDVersion ModelLoader::get_sd_version() const {
TensorStorage token_embedding_weight, input_block_weight, context_ebedding_weight;
if (tensor_storage_map.find("model.diffusion_model.emb_in.image.weight") != tensor_storage_map.end() &&
tensor_storage_map.find("model.diffusion_model.txt_mode_blocks.0.q_proj.weight") != tensor_storage_map.end()) {
return VERSION_FLUX3;
}
bool has_multiple_encoders = false;
bool is_unet = false;