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