A client library for the Stability AI SDK.
Requires an API Key for DreamStudio which can be found here.
# NPM
npm i -g stability-client
# To Update
npm update -g stability-client
# Yarn
yarn global add stability-clientstability \
-s 150 \ # Steps
-c 15 \ # Cfg Scale
-a k_euler_ancestral \ # Diffusion Method
-S 3465383516 \ # Seed
-o ./examples \ # Output Directory
-n 5 \ # 5 images
"a anime still of an highly detailed night cyberpunk city life, bladerunner style!! detailed shops, neon lights, ray tracing, advertising everywhere, people and robots walking around. art by satoshi kon and studio ghibli, in the style of ghost in the shell, muted colours, hyperrealism, cinematic lighting, lush detail, award winning, wlop, octane render, trending on artstation"import { generate } from 'stability-client'
const api = generate({
prompt: 'A Stunning House',
apiKey: process.env.DREAMSTUDIO_API_KEY,
})
api.on('image', ({ buffer, filePath }) => {
console.log('Image', buffer, filePath)
})
api.on('end', (data) => {
console.log('Generating Complete', data)
})Async/Promise API
import { generateAsync } from 'stability-client'
try {
const { res, images } = await generateAsync({
prompt: 'A Stunning House',
apiKey: process.env.DREAMSTUDIO_API_KEY,
})
console.log(images)
} catch (e) {
// ...
}const imgBuffer = Buffer.from(base64ImageRaw, 'base64')
const api = await generate({
prompt: `amazing looking room, Dean Norton style`,
apiKey: process.env.STABLE_DIFFUSION_KEY,
width: 512,
height: 512,
steps: 10,
engine: 'stable-diffusion-512-v2-1',
cfgScale: 10,
noStore: false, // if set to true, it won't save files to outDir after generation.
imagePrompt: {
mime: 'image/png',
content: imgBuffer,
},
samples: 1,
diffusion: 'ddim',
outDir: '/output',
})
api.on('image', async ({ buffer, imagePath }) => {
// upload somewhere probably..
})
api.on('end', (data) => {
console.log('Generating Complete', data)
})Usage: stability [options] [prompt]
Generate an image
Arguments:
prompt The text prompt you want to use
Options:
-V, --version output the version number
-H, --height <height> height of image (default: 512)
-W, --width <width> width of image (default: 512)
-c, --cfg_scale <scale> CFG scale factor (default: 7)
-a, --sampler <sampler> Diffusion Method (choices: "ddim", "plms", "k_euler", "k_euler_ancestral", "k_heun", "k_dpm_2", "k_dpm_2_ancestral", "k_lms", default: "k_lms")
-s, --steps <steps> number of steps (default: 50)
-S, --seed <seed> random seed to use (default: 1614811539)
-n, --num_samples <num_samples> number of samples to generate (default: 1)
-e, --engine <engine> engine to use for inference (default: "stable-diffusion-v1")
--no-store do not write artifacts to disk
-k, --api-key <api-key> DreamStudio API Key (env: DREAMSTUDIO_API_KEY)
-o, --output-dir <outputDir> directory to store images (defaults to cwd)
-d, --debug Additional logging
-h, --help display help for commandnvm use
yarn
yarn build
npm link
export DREAMSTUDIO_API_KEY=...
stability "A stunning house"