Skip to content

Commit 9195fed

Browse files
committed
fix(diffusers): add support for diffusers SDXL text encoder prefixes
Problem: SDXL models in diffusers directory format fail to load with "unknown tensor" errors Solution: Added te. and te.1. prefixes to cond_stage_model conversion list Testing: SDXL diffusers models now load and generate successfully Root cause: When loading diffusers SDXL models, text_encoder uses "te." prefix and text_encoder_2 uses "te.1." prefix. These weren't in the name conversion prefix list, so tensors weren't being converted to checkpoint format names. This fix enables diffusers-format SDXL models to work alongside single-file checkpoint models without requiring format conversion. Fixes: Models like duchaiten-pony-real-v20-sdxl in diffusers directory layout
1 parent f454181 commit 9195fed

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/name_conversion.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -920,6 +920,8 @@ std::vector<std::string> cond_stage_model_prefix_vec = {
920920
"cond_stage_model.",
921921
"conditioner.embedders.",
922922
"text_encoders.",
923+
"te.1.", // diffusers SDXL text_encoder_2 (clip_g)
924+
"te.", // diffusers text_encoder (clip_l)
923925
};
924926

925927
std::vector<std::string> diffuison_model_prefix_vec = {

0 commit comments

Comments
 (0)