【WebUI 1111】構図ガチャ用スクリプト AutoSkipの使い方【Stable diffusion】

 低ステップで構図ガチャをして良いのが出来ても、ステップ数を変更すると構図が変わるので、途中で止めるスクリプトを作りました。

インストール

下の AutoSkip.py をクリックして、保存したファイルを scriptsフォルダーに移動して Webuiを再起動します。

AutoSkip.py
#AutoSkip ver 1.4 (c)2023 yukitoki.blogspot.com #1.4 i2iの時％で設定した時、ステップ数が半分になる様にしました。 #1.3 t2iの後i2iをするとt2iの設定になる不具合を修正 #1.2 pnginfoへの記録方法を変更しました。 #1.1 pnginfoに情報が複数記録される不具合を修正 import modules.scripts as scripts import gradio as gr import math from modules.shared import state from modules.script_callbacks import CFGDenoisedParams, on_cfg_denoised,ImageSaveParams, on_before_image_saved, AfterCFGCallbackParams, on_cfg_after_cfg, on_infotext_pasted from modules import script_callbacks NAME = 'AutoSkip' _skipstep = 0 class Script(scripts.Script): _skipmode = "Off" _myskip = False _img2img = False def title(self): return NAME def show(self, is_img2img): return scripts.AlwaysVisible def ui(self, is_img2img): with gr.Accordion(NAME, open=True): skipmode = gr.Radio(["Off","80%","60%","50%","40%","Custom"], value="Off", label="Skip Mode") skipstep = gr.Slider(minimum=0, maximum=30, step=1, value=0, label='Custom Step') self._img2img = is_img2img return [skipmode,skipstep] def denoised_callback(self, params: CFGDenoisedParams): global _skipstep if _skipstep > 0: if params.sampling_step >= _skipstep - 2: state.skip() self._myskip = True def process(self, p,skipmode, skipstep): global _skipstep self._skipmode = getattr(p, 'autoskip_skipmode', skipmode) _skipstep = getattr(p, 'autoskip_skipstep', skipstep) if(self._skipmode == 'Off'): _skipstep = 0 elif(self._skipmode == '80%'): _skipstep = math.floor(p.steps * 0.8) elif(self._skipmode == '60%'): _skipstep = math.floor(p.steps * 0.6) elif(self._skipmode == '50%'): _skipstep = math.floor(p.steps * 0.5) elif(self._skipmode == '40%'): _skipstep = math.floor(p.steps * 0.4) if(self._skipmode != 'Custom' and self._img2img): _skipstep = math.floor(_skipstep * 0.5) if(_skipstep > 0): print('Auto Skip:',_skipstep,'Step') p.extra_generation_params["Auto Skip Step"] = _skipstep else: if("Auto Skip Step" in p.extra_generation_params.keys()): del p.extra_generation_params["Auto Skip Step"] on_cfg_denoised(self.denoised_callback) return def process_batch(self, p, *args, **kwargs): self._myskip = False def postprocess_batch(self, p, *args, **kwargs): if(self._myskip): state.skipped = False # xyz support def make_axis_on_xyz_grid(): xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "xyz_grid.py"][0].module extra_axis_options = [ xyz_grid.AxisOption("["+NAME+"] Skip Mode", str, xyz_grid.apply_field("autoskip_skipmode"), choices=lambda:['Off','80%','60%','50%','40%','Custom']), xyz_grid.AxisOption("["+NAME+"] Custom Step", int,xyz_grid.apply_field("autoskip_skipstep")), ] if not any("["+NAME+"]" in x.label for x in xyz_grid.axis_options): xyz_grid.axis_options.extend(extra_axis_options) def on_before_ui(): try: make_axis_on_xyz_grid() except Exception: error = traceback.format_exc() print( f" - Auto Skip : xyz_grid error:\n{error}", file=sys.stderr, ) script_callbacks.on_before_ui(on_before_ui)

 

使い方

 インストールが出来ると Extension に AutoSkipタブ が追加されます。

Skip Modeを選択して止めたい位置を決めます。

Customを選択すると、Custom Stepで指定したステップ数で止める事が出来ます。

0の時は、Offになります。 

具体的には、ステップ数を最後に出したい数値にして、SkipModeを５０％辺りで構図ガチャします。

良いのが出たらOffにして、ガチャした時と同じステップ数で本番用を作成する感じで使います。

本番用の作成には、txt2imgでバッチ処理ができる「 SD-webui-txt2img-script 」を使うと便利です。

参考画像

２０ステップを設定してAutoSkipで１０ステップで止めた時と、実際の１０ステップとの違い

低ステップだと使えないSDE系でもガチャできるかも！？

DPM++ 2M SDE Exponentialにdiscard_next_to_last_sigmaを付けたサンプラーで、２０ステップを１０ステップで止めた時の絵。これはこれで、ふわっとしてて良い感じがする。