theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-25 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug: textual inversion

Browse Source
This commit is contained in:
Artiprocher
2024-01-09 20:33:17 +08:00
parent e6f9156b14
commit 552355692d
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/models/__init__.py
View File

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