【Stable diffusion】 AUTOMATIC1111で800枚以上無制限に生成するCustom Scriptを導入

2023年7月23日2023年7月26日

Stable DiffusionをAUTOMATIC1111で使用している場合，Batch countとBatch sizeを利用することで一度に最大800枚の画像を生成することができます．

「800枚じゃ物足りない」「寝ている間にもっと出力したい」などの要求がある場合は，別の方法を検討する必要があります．

この記事では，枚数ほぼ無制限に出力することができる”Custom Script”の導入方法を示します．

Custom Scripts

Custom Scriptsは，AUTOMATIC1111が用意しているPython用のスクリプトツールです．

modules.scriptを継承してプログラムを記述し，AUTOMATIC1111で実行します．

N回実行するプログラムが下記の通りです．

run_n_times.py


import math
import os
import sys
import traceback

import modules.scripts as scripts
import gradio as gr

from modules.processing import Processed, process_images

class Script(scripts.Script):
    def title(self):
        return "Run n times"

    def ui(self, is_img2img):
        n = gr.Textbox(label="n")
        return [n]

    def run(self, p, n):
        for x in range(int(n)):
            p.seed = -1
            proc = process_images(p)
            image = proc.images
        return Processed(p, image, p.seed, proc.info)