1 unstable release
0.1.0 | Aug 25, 2023 |
---|
#26 in #text-image
58KB
1K
SLoC
stabilityai
Rust library for stability.ai
Overview
stabilityai
is an unofficial Rust library for stability.ai
- It's based on OpenAPI spec
- Current features:
- Users
- Engines
- Generation
The library reads API key from the environment variable STABILITY_API_KEY
.
# On macOS/Linux
export STABILITY_API_KEY='sk-...'
# On Windows Powershell
$Env:STABILITY_API_KEY='sk-...'
- Visit examples for full examples.
- Visit docs.rs/stabilityai for docs.
Text To Image Example
use stabilityai::{
error::StabilityAIError,
types::{ClipGuidancePreset, Sampler, StylePreset, TextToImageRequestBodyArgs},
Client,
};
use std::error::Error;
#[tokio::main]
async fn main() -> Result<(), Box<dyn Error>> {
// create client, reads STABILITY_API_KEY environment variable for API key.
let client = Client::new();
let request = TextToImageRequestBodyArgs::default()
.text_prompts(
"A banner with ocean background having a cute \
red cartoon crab on beach and boats in the ocean",
)
.samples(1)
.steps(30_u32)
.clip_guidance_preset(ClipGuidancePreset::FastBlue)
.sampler(Sampler::KDpmpp2sAncestral)
.width(1216_u16)
.height(832_u16)
.style_preset(StylePreset::ThreeDModel)
.build()?;
let artifacts = client
.generate("stable-diffusion-xl-1024-v1-0")
.text_to_image(request)
.await?;
// Create directory if doesn't exist and save images
let paths = artifacts.save("./data").await?;
paths
.iter()
.for_each(|path| println!("Image saved at {}", path.display()));
Ok(())
}
Contributing
Thank you for your time to contribute and improve the project, I'd be happy to have you!
A good starting point cloud be an existing open issue.
License
This project is licensed under MIT license.
Dependencies
~7–21MB
~344K SLoC