mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-07 15:52:32 +08:00
First step of supporting save filenames without trailing _ (#13722)
get_save_image_path now properly supports filenames without trailing underscores. This will be the saving behavior when using a mix of save image nodes using the old and the new format. ComfyUI_00001_.png ComfyUI_00002.png ComfyUI_00003.png ComfyUI_00004_.png
This commit is contained in:
parent
e5369c0eec
commit
c168960a12
@ -432,7 +432,9 @@ def get_save_image_path(filename_prefix: str, output_dir: str, image_width=0, im
|
|||||||
prefix_len = len(os.path.basename(filename_prefix))
|
prefix_len = len(os.path.basename(filename_prefix))
|
||||||
prefix = filename[:prefix_len + 1]
|
prefix = filename[:prefix_len + 1]
|
||||||
try:
|
try:
|
||||||
digits = int(filename[prefix_len + 1:].split('_')[0])
|
remainder = filename[prefix_len + 1:]
|
||||||
|
base_remainder = remainder.split('.')[0]
|
||||||
|
digits = int(base_remainder.split('_')[0])
|
||||||
except:
|
except:
|
||||||
digits = 0
|
digits = 0
|
||||||
return digits, prefix
|
return digits, prefix
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user