muerrilla/sd-webui-detail-daemon
1
0
Fork 0
mirror of https://github.com/muerrilla/sd-webui-detail-daemon.git synced 2026-01-26 11:19:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4 from w-e-w/main

Browse Source 
fix disable during hires fix
This commit is contained in:
Muerrilla
2024-07-01 22:13:22 +03:30
committed by GitHub
parent 9621682aee f1a3c5ed8f
commit 6f38ff3e9f
1 changed files with 21 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
scripts/detail_daemon.py
View File

@@ -114,8 +114,8 @@ class Script(scripts.Script):
fade = getattr(p, "DD_fade", fade)
smooth = getattr(p, "DD_smooth", smooth)
if enabled :
if p.sampler_name == "DPM adaptive" :
if enabled:
if p.sampler_name == "DPM adaptive":
tqdm.write(f'\033[33mINFO:\033[0m Detail Daemon does not work with {p.sampler_name}')
return
# Restart can be handled better, later maybe
@@ -123,7 +123,6 @@ class Script(scripts.Script):
actual_steps = (p.steps * 2 - 1) if p.sampler_name in ['DPM++ SDE', 'DPM++ 2S a', 'Heun', 'DPM2', 'DPM2 a', 'Restart'] else p.steps
self.schedule = self.make_schedule(actual_steps, start, end, bias, amount, exponent, start_offset, end_offset, fade, smooth)
self.mode = mode
self.is_hires = False
self.cfg_scale = p.cfg_scale
on_cfg_denoiser(self.denoiser_callback)
self.callback_added = True
@@ -135,6 +134,9 @@ class Script(scripts.Script):
delattr(self, 'callback_added')
# tqdm.write('\033[90mINFO: Detail Daemon callback removed\033[0m')
def before_process_batch(self, p, *args, **kwargs):
self.is_hires = False
def postprocess(self, p, processed, *args):
if hasattr(self, 'callback_added'):
remove_callbacks_for_function(self.denoiser_callback)
@@ -148,15 +150,15 @@ class Script(scripts.Script):
tqdm.write(f'\033[33mINFO:\033[0m Detail Daemon does not work during Hires Fix')
def denoiser_callback(self, params):
if self.is_hires :
if self.is_hires:
return
idx = params.denoiser.step
multiplier = 1 - self.schedule[idx]
if self.mode == "cond" :
if self.mode == "cond":
params.sigma[0] *= 1 - self.schedule[idx] * .1
elif self.mode == "uncond" :
elif self.mode == "uncond":
params.sigma[1] *= 1 - self.schedule[idx] * -.1
else :
else:
params.sigma *= 1 - self.schedule[idx] * .1 * self.cfg_scale
def make_schedule(self, steps, start, end, bias, amount, exponent, start_offset, end_offset, fade, smooth):
@@ -196,7 +198,8 @@ class Script(scripts.Script):
values = self.make_schedule(steps, start, end, bias, amount, exponent, start_offset, end_offset, fade, smooth)
mean = sum(values)/steps
peak = np.clip(max(abs(values)), -1, 1)
if start > end : start = end
if start > end:
start = end
mid = start + bias * (end - start)
opacity = .1 + (1 - fade) * 0.7
plot_color = (0.5, 0.5, 0.5, opacity) if not enabled else ((1 - peak)**2, 1, 0, opacity) if mean >= 0 else (1, (1 - peak)**2, 0, opacity)
@@ -210,23 +213,24 @@ class Script(scripts.Script):
"ytick.labelcolor": plot_color,
"ytick.color": plot_color,
})
fig, ax = plt.subplots(figsize=(2.15, 2.00),layout="constrained")
fig, ax = plt.subplots(figsize=(2.15, 2.00), layout="constrained")
ax.plot(range(steps), values, color=plot_color)
ax.axhline(y=0, color=plot_color, linestyle='dotted')
ax.axvline(x=mid * (steps - 1), color=plot_color, linestyle='dotted')
ax.tick_params(right=False, color=plot_color)
ax.set_xticks([i * (steps - 1) / 10 for i in range(10)][1:])
ax.set_xticklabels([])
ax.set_ylim([-1,1])
ax.set_xlim([0,steps-1])
ax.set_ylim([-1, 1])
ax.set_xlim([0, steps-1])
plt.close()
self.last_vis = fig
return fig
except:
if self.last_vis is not None :
return fig
except Exception:
if self.last_vis is not None:
return self.last_vis
return
def xyz_support():
for scriptDataTuple in scripts.scripts_data:
if os.path.basename(scriptDataTuple.path) == 'xyz_grid.py':
@@ -234,7 +238,7 @@ def xyz_support():
def confirm_mode(p, xs):
for x in xs:
if x not in ['both','cond', 'uncond']:
if x not in ['both', 'cond', 'uncond']:
raise RuntimeError(f'Invalid Detail Daemon Mode: {x}')
mode = xy_grid.AxisOption(
'[Detail Daemon] Mode',
@@ -293,6 +297,8 @@ def xyz_support():
end_offset,
fade,
])
try:
xyz_support()
except Exception as e: