This is an simple python SDK and CLI for Astria.ai, the DreamBooth based image generation service.
When used as a command line tool, these are the arguments:
gen Generate images for prompts
tune Create a new tune
list List all generated images for a tune
listtune List all tunes in the account
info Get details of prompt
download Download images of prompt
downloadall Download recent images
Optional arguments:
- --key KEY Astria API token (if unspecified, token is taken from the enviroment variable ASTRIA_API_TOKEN)
- --outdir OUTDIR Output directory for images (default is current working directory)
python astria.py gen [--seed SEED] [--steps STEPS] [--download] tuneid prompts [prompts ...]
- --seed SEED
- --steps STEPS
- --download Download results
python astria.py tune [--test] title classname images [images ...]
title Descriptive name for this tuning
classname Dreambooth classname (style, person, man, woman etc)
images At least 10 image filenames
python astria.py list tuneid
python astria.py listtune
python astria.py info tuneid promptid
- tuneid Astria Tune ID
- promptid Prompt ID
python astria.py download [--wait] tuneid promptid
-
tuneid Astria Tune ID
-
promptid Prompt ID
optional arguments:
-
--wait Wait for results to be ready
python astria.py downloadall [--limit LIMIT] [--db DB] [tuneids ...]
-
tuneids One or more Astria Tune IDs, or emtpy to download entire account
optional arguments:
-
--limit LIMIT Maximum number of prompts to download from each fine tune
-
--db DB Keep list of previously downloaded images in a DB file