浏览代码

add embeddings dir

AUTOMATIC 2 年之前
父节点
当前提交
98cc6c6e74
共有 2 个文件被更改,包括 6 次插入1 次删除
  1. 0 0
      embeddings/Place Textual Inversion embeddings here.txt
  2. 6 1
      modules/sd_hijack.py

+ 0 - 0
embeddings/Place Textual Inversion embeddings here.txt


+ 6 - 1
modules/sd_hijack.py

@@ -232,7 +232,12 @@ class StableDiffusionModelHijack:
 
         for fn in os.listdir(dirname):
             try:
-                process_file(os.path.join(dirname, fn), fn)
+                fullfn = os.path.join(dirname, fn)
+
+                if os.stat(fullfn).st_size == 0:
+                    continue
+
+                process_file(fullfn, fn)
             except Exception:
                 print(f"Error loading emedding {fn}:", file=sys.stderr)
                 print(traceback.format_exc(), file=sys.stderr)