1 unstable release
0.1.0 | Sep 27, 2023 |
---|
#613 in Machine learning
72KB
1.5K
SLoC
An unofficial Rust API client for Stability-AI text-to-speech software.
⚙️ Requirements
- Set API key to environment variable
STABILITY_API_KEY
🗣️ Usage
Text to Image
use stability_rs::{text_to_img::*, Result, ClipGuidancePreset, Sampler, StylePreset};
#[tokio::main]
async fn main() -> Result<()> {
let image = TextToImageBuilder::new()
.height(1024)?
.width(1024)?
.cfg_scale(27)?
.clip_guidance_preset(ClipGuidancePreset::FastBlue)?
.sampler(Sampler::KDpmpp2sAncestral)?
.samples(2)?
.seed(0)?
.steps(33)?
.style_preset(StylePreset::DigitalArt)?
.text_prompt("A scholar tired at his desk, a raven on a bust", 1.0)?
.build()?;
let resp = image.generate("stable-diffusion-xl-1024-v1-0").await?;
for (i, image) in resp.artifacts.iter().enumerate() {
let _ = image.save(&format!("image_{}.png", i)).await?;
}
Ok(())
}
Image to Image
use stability_rs::{img_to_img::*, Result, ClipGuidancePreset, Sampler, StylePreset,};
#[tokio::main]
async fn main() -> Result<()> {
let image = ImageToImageBuilder::new()
.init_image_path("init_image.png")?
.init_image_mode(ImageMode::ImageStrength)?
.image_strength(0.35)?
.cfg_scale(7)?
.clip_guidance_preset(ClipGuidancePreset::FastBlue)?
.sampler(Sampler::KDpm2Ancestral)?
.samples(3)?
.seed(0)?
.steps(20)?
.style_preset(StylePreset::FantasyArt)?
.text_prompt("A crab relaxing on a beach", 0.5)?
.text_prompt("stones", -0.9)?
.build()?;
let resp = image.generate("stable-diffusion-xl-1024-v1-0").await?;
for (i, img) in resp.artifacts.iter().enumerate() {
let _ = img.save(&format!("new_image_{}.png", i)).await?;
}
Ok(())
}
Image Upscaling
use stability_rs::{upscale::*, Result,};
#[tokio::main]
async fn main() -> Result<()> {
let image = UpscalerBuilder::new()
.image("1024_image.png")?
.height(2048)?
.build()?;
let resp = image.generate(UpscaleEngine::EsrganV1X2Plus).await?;
resp.artifacts.first().unwrap().save("2048_image.png").await?;
Ok(())
}
Image Masking
use stability_rs::{masking::*, Result, StylePreset, ClipGuidancePreset};
#[tokio::main]
async fn main() -> Result<()> {
let engine = "stable-inpainting-512-v2-0";
let image = MaskerBuilder::new()
.init_image_path("init_image.png")?
.mask_source(MaskSource::MaskImageBlack)?
.mask_image("black_mask_image.png")?
.text_prompt("a crab dancing", 1.0)?
.style_preset(StylePreset::FantasyArt)?
.clip_guidance_preset(ClipGuidancePreset::FastBlue)?
.build()?;
let resp = image.generate(engine).await?;
resp.artifacts.first().unwrap().save("masked_image.png").await?;
Ok(())
}
Dependencies
~8–19MB
~269K SLoC