Code to Generate Prompts and Seeds from Txt Files

[grexzen]

Sharing this script that I am using to do large batches (10,000 generations) for research by adding multiple prompts and seeds from txt files. Can be used with the custom code script.

I was going to do a script, but the last update even broke SD Chad, so just sharing quickly in case others find it useful.

import modules.scripts as scripts
import gradio as gr
import os
import itertools

from modules import images
from modules.processing import process_images, Processed
from modules.processing import Processed
from modules.shared import opts, cmd_opts, state

filec = open("/content/captions.txt") #ADD YOUR PROMPTS FILENAME PATH HERE
files = open("/content/seeds.txt") #ADD YOUR SEEDS FILENAME PATH HERE

datac = filec.read()
datas = files.read()

captions = datac.split("\n")
seeds = datas.split("\n")

def run(p, captions, seeds):
      for (caption, seed) in zip(captions, seeds):
            p.prompt = caption
            #print(caption)
            p.seed = seed
            #print(seed)
            proc = process_images(p)
            gens = proc.images
      return Processed(p, gens, p.prompt, p.seed, "")

run(p, captions, seeds)