Skip to content

Commit

Permalink
Revert "Merge pull request #2410 from vladmandic/master"
Browse files Browse the repository at this point in the history
This reverts commit 597fc18, reversing
changes made to 4ab584e.
  • Loading branch information
vladmandic committed Oct 26, 2023
1 parent 64cce8a commit 886af2e
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 31 deletions.
1 change: 0 additions & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,6 @@ Mostly service release with support for several new models and additional optimi
- fix diffusers samplers: dpm++ 2m, dpm++ 1s, deis
- fix new style filename template
- fix image name template using model name
- fix image name sequence
- fix model path using relative path
- fix `torch-rocm` and `tensorflow-rocm` version detection, thanks @xangelix
- fix **chainner** upscalers color clipping
Expand Down
2 changes: 1 addition & 1 deletion extensions-builtin/sd-webui-controlnet
37 changes: 16 additions & 21 deletions modules/images.py
Original file line number Diff line number Diff line change
Expand Up @@ -553,23 +553,33 @@ def save_image(image, path, basename = '', seed=None, prompt=None, extension=sha
return None, None
if not check_grid_size([image]):
return None, None
if path is None or len(path) == 0:
if path is None or len(path) == 0: # set default path to avoid errors when functions are triggered manually or via api and param is not set
path = shared.opts.outdir_save

# namegen
namegen = FilenameGenerator(p, seed, prompt, image, grid=grid)
if shared.opts.save_to_dirs:
dirname = namegen.apply(shared.opts.directories_filename_pattern or "[date]")
dirname = namegen.apply(shared.opts.directories_filename_pattern or "[prompt_words]")
path = os.path.join(path, dirname)
if forced_filename is None:
if short_filename or seed is None:
file_decoration = ""
if shared.opts.samples_filename_pattern and len(shared.opts.samples_filename_pattern) > 0:
file_decoration = shared.opts.samples_filename_pattern
else:
file_decoration = "[seq]-[model_name]-[prompt_words]"
file_decoration = "[seq]-[prompt_words]"
file_decoration = namegen.apply(file_decoration)
filename = os.path.join(path, f"{file_decoration}{suffix}.{extension}") if basename is None or basename == '' else os.path.join(path, f"{basename}-{file_decoration}{suffix}.{extension}")
file_decoration += suffix
if shared.opts.save_images_add_number:
if '[seq]' not in file_decoration:
file_decoration = f"[seq]-{file_decoration}"
basecount = get_next_sequence_number(path, basename)
filename = None
for i in range(9999):
seq = f"{basecount + i:05}" if basename == '' else f"{basename}-{basecount + i:04}"
filename = os.path.join(path, f"{file_decoration.replace('[seq]', seq)}.{extension}")
if not os.path.exists(filename):
break
else:
filename = os.path.join(path, f"{file_decoration}.{extension}") if basename == '' else os.path.join(path, f"{basename}-{file_decoration}.{extension}")
else:
filename = os.path.join(path, f"{forced_filename}.{extension}")
pnginfo = existing_info or {}
Expand All @@ -579,21 +589,6 @@ def save_image(image, path, basename = '', seed=None, prompt=None, extension=sha
params.filename = namegen.sanitize(filename)
dirname = os.path.dirname(params.filename)
os.makedirs(dirname, exist_ok=True)

# sequence
if shared.opts.save_images_add_number or '[seq]' in params.filename:
if '[seq]' not in params.filename:
params.filename = f"[seq]-{params.filename}"
basecount = get_next_sequence_number(dirname, basename)
for i in range(9999):
seq = f"{basecount + i:05}" if basename == '' else f"{basename}-{basecount + i:04}"
filename = params.filename.replace('[seq]', seq)
if not os.path.exists(filename):
debug(f'Prompt sequence: input="{params.filename}" seq={seq} output="{filename}"')
params.filename = filename
break

# callbacks
script_callbacks.before_image_saved_callback(params)
exifinfo = params.pnginfo.get('UserComment', '')
exifinfo = (exifinfo + ', ' if len(exifinfo) > 0 else '') + params.pnginfo.get(pnginfo_section_name, '')
Expand Down
8 changes: 0 additions & 8 deletions modules/processing.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,14 +72,6 @@ def apply_overlay(image, paste_loc, index, overlays):
return image


def create_binary_mask(image):
if image.mode == 'RGBA' and image.getextrema()[-1] != (255, 255):
image = image.split()[-1].convert("L").point(lambda x: 255 if x > 128 else 0)
else:
image = image.convert('L')
return image


def txt2img_image_conditioning(sd_model, x, width, height):
if sd_model.model.conditioning_key in {'hybrid', 'concat'}: # Inpainting models
# The "masked-image" in this case will just be all zeros since the entire image is masked.
Expand Down

0 comments on commit 886af2e

Please sign in to comment.