openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-14 22:53:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

add safety check in case of short extensions

Browse Source 
so eg if a two-letter or empty extension is used, `.txt` would break, this `max` call protects that.
This commit is contained in:
Alex "mcmonkey" Goodwin 2023-03-12 12:33:29 -07:00
parent a71b7b5ec0
commit 48df6d66ea
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/images.py
View File

@ -575,7 +575,7 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
fullfn_without_extension, extension = os.path.splitext(params.filename)
if hasattr(os, 'statvfs'):
max_name_len = os.statvfs(path).f_namemax
fullfn_without_extension = fullfn_without_extension[:max_name_len - len(extension)]
fullfn_without_extension = fullfn_without_extension[:max_name_len - max(4, len(extension))]
params.filename = fullfn_without_extension + extension
_atomically_save_image(image, fullfn_without_extension, extension)