StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix two bug

Browse Source
This commit is contained in:
yfszzx 2022-10-17 23:14:03 +08:00
parent 2272cf2f35
commit 2b5b62e768
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/images_history.py
View File

@ -44,7 +44,7 @@ def traverse_all_files(curr_path, image_list, all_type=False):
return image_list return image_list
for file in f_list: for file in f_list:
file = os.path.join(curr_path, file) file = os.path.join(curr_path, file)
if (not all_type) and file[-4:] == ".txt": if (not all_type) and (file[-4:] == ".txt" or file[-4:] == ".csv"):
pass pass
elif os.path.isfile(file) and file[-10:].rfind(".") > 0: elif os.path.isfile(file) and file[-10:].rfind(".") > 0:
image_list.append(file) image_list.append(file)
@ -182,7 +182,7 @@ def delete_image(delete_num, name, filenames, image_index, visible_num):
def save_image(file_name): def save_image(file_name):
if file_name is not None and os.path.exists(file_name): if file_name is not None and os.path.exists(file_name):
shutil.copy2(file_name, opts.outdir_save) shutil.copy(file_name, opts.outdir_save)
def get_recent_images(page_index, step, filenames): def get_recent_images(page_index, step, filenames):
page_index = int(page_index) page_index = int(page_index)