fix bug: textual inversion

2026-03-18 22:08:13 +00:00 · 2024-01-09 20:33:17 +08:00
parent e6f9156b14
commit 552355692d
1 changed files with 2 additions and 0 deletions
--- a/diffsynth/models/init.py
+++ b/diffsynth/models/init.py
@@ -153,6 +153,8 @@ class ModelManager:

        # Load every textual inversion file
        for file_name in os.listdir(folder):
+            if file_name.endswith(".txt"):
+                continue
            keyword = os.path.splitext(file_name)[0]
            state_dict = load_state_dict(os.path.join(folder, file_name))