Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

unconditional options init

Browse Source
This commit is contained in:
Vladimir Mandic
2023-07-14 08:49:17 -04:00
parent 558b71f088
commit 7bb6c213de
7 changed files with 42 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/xyz_grid.py
+5 -5
View File
@@ -131,8 +131,8 @@ def apply_fallback(p, x, xs):
shared.opts.data["force_latent_sampler"] = sampler_name
def apply_uni_pc_order(p, x, xs):
shared.opts.data["uni_pc_order"] = min(x, p.steps - 1)
def apply_schedulers_solver_order(p, x, xs):
shared.opts.data["schedulers_solver_order"] = min(x, p.steps - 1)
def apply_face_restore(p, opt, x):
@@ -231,7 +231,7 @@ axis_options = [
AxisOption("Sampler Eta", float, apply_field("eta")),
AxisOptionTxt2Img("Hires upscaler", str, apply_field("hr_upscaler"), choices=lambda: [*shared.latent_upscale_modes, *[x.name for x in shared.sd_upscalers]]),
AxisOptionImg2Img("Image Mask Weight", float, apply_field("inpainting_mask_weight")),
AxisOption("UniPC Order", int, apply_uni_pc_order, cost=0.5),
AxisOption("Sampler Solver Order", int, apply_schedulers_solver_order, cost=0.5),
AxisOption("Face restore", str, apply_face_restore, fmt=format_value),
AxisOption("Token merging ratio", float, apply_override('token_merging_ratio')),
AxisOption("Token merging ratio high-res", float, apply_override('token_merging_ratio_hr')),
@@ -345,7 +345,7 @@ class SharedSettingsStackHelper(object):
def __enter__(self):
#Save overridden settings so they can be restored later.
self.vae = shared.opts.sd_vae
self.uni_pc_order = shared.opts.uni_pc_order
self.schedulers_solver_order = shared.opts.schedulers_solver_order
self.token_merging_ratio_hr = shared.opts.token_merging_ratio_hr
self.token_merging_ratio = shared.opts.token_merging_ratio
self.sd_model_checkpoint = shared.opts.sd_model_checkpoint
@@ -356,7 +356,7 @@ class SharedSettingsStackHelper(object):
def __exit__(self, exc_type, exc_value, tb):
#Restore overriden settings after plot generation.
shared.opts.data["sd_vae"] = self.vae
shared.opts.data["uni_pc_order"] = self.uni_pc_order
shared.opts.data["schedulers_solver_order"] = self.schedulers_solver_order
shared.opts.data["token_merging_ratio_hr"] = self.token_merging_ratio_hr
shared.opts.data["token_merging_ratio"] = self.token_merging_ratio
shared.opts.data["force_latent_sampler"] = self.force_latent_sampler