Stable-Diffusion Productivity Scripts

Note: All scripts have built-in --help parameter that can be used to get more information

Main Scripts

Generate

Text-to-image with all of the possible parameters
Supports upsampling, face restoration and grid creation

python generate.py

By default uses parameters from generate.json

Parameters that are not specified will be randomized:

Prompt will be dynamically created from template of random samples: random.json
Sampler/Scheduler will be randomly picked from available ones
CFG Scale set to 5-10

Train

Combined pipeline for embeddings, lora, lycoris, dreambooth and hypernetwork
Optionally runs several image processing steps before training:

keep original image
detect and extract face
detect and extract body
detect blur
detect dynamic range
attempt to upscale low resolution images
attempt to restore quality of low quality images
automatically generate captions using interrogate
resize image
square image
run image segmentation to remove background

python train.py

Auxiliary Scripts

Benchmark

python run-benchmark.py

Create Previews

Create previews for embeddings, lora, lycoris, dreambooth and hypernetwork

python create-previews.py

Image Grid

python image-grid.py

Image Watermark

Create invisible image watermark and remove existing EXIF tags

python image-watermark.py

Image Interrogate

Runs CLiP and Booru image interrogation

python image-interrogate.py

Palette Extract

Extract color palette from image(s)

python image-palette.py

Prompt Ideas

Generate complex prompt ideas

python prompt-ideas.py

Prompt Promptist

Attempts to beautify the provided prompt

python prompt-promptist.py

Video Extract

Extract frames from video files

python video-extract.py

Utility Scripts

SDAPI

Utility module that handles async communication to Automatic API endpoints
Note: Requires SD API

Can be used to manually execute specific commands:

python sdapi.py progress
python sdapi.py interrupt python sdapi.py shutdown