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: true, //means it will not store in the .out dir 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 aritfacts 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"
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent