uniform listdir and use threadpool to index loras

This commit is contained in:
Vladimir Mandic
2024-01-06 15:15:23 -05:00
parent dd7e3e93c4
commit de80b74c64
20 changed files with 64 additions and 114 deletions
+1 -1
View File
@@ -38,7 +38,7 @@ def run_postprocessing(extras_mode, image, image_folder: List[tempfile.NamedTemp
elif extras_mode == 2:
assert not shared.cmd_opts.hide_ui_dir_config, '--hide-ui-dir-config option must be disabled'
assert input_dir, 'input directory not selected'
image_list = shared.listfiles(input_dir)
image_list = shared.listdir(input_dir)
for filename in image_list:
try:
image = Image.open(filename)