Skip to content
This repository has been archived by the owner on Jun 28, 2024. It is now read-only.

Latest commit

 

History

History
108 lines (64 loc) · 2.14 KB

README.md

File metadata and controls

108 lines (64 loc) · 2.14 KB

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