sd

2026-04-24 06:46:13 +00:00 · 2026-04-23 17:35:24 +08:00
parent 5c89a15b9a
commit 82e482286c
14 changed files with 3274 additions and 1 deletions
--- a/diffsynth/utils/state_dict_converters/stable_diffusion_text_encoder.py
+++ b/diffsynth/utils/state_dict_converters/stable_diffusion_text_encoder.py
@@ -0,0 +1,7 @@
+def SDTextEncoderStateDictConverter(state_dict):
+    new_state_dict = {}
+    for key in state_dict:
+        if key.startswith("text_model.") and "position_ids" not in key:
+            new_key = "model." + key
+            new_state_dict[new_key] = state_dict[key]
+    return new_state_dict