stabilityai

Rust library for stability.ai

Overview

stabilityai is an unofficial Rust library for stability.ai

• It's based on OpenAPI spec
• Current features:
  • [x] Users
  • [x] Engines
  • [x] Generation

The library reads API key from the environment variable STABILITY_API_KEY.

```bash

On macOS/Linux

export STABILITYAPIKEY='sk-...' ```

```powershell

On Windows Powershell

$Env:STABILITYAPIKEY='sk-...' ```

• Visit examples for full examples.
• Visit docs.rs/stabilityai for docs.

Text To Image Example

```rust use stabilityai::{ error::StabilityAIError, types::{ClipGuidancePreset, Sampler, StylePreset, TextToImageRequestBodyArgs}, Client, }; use std::error::Error;

[tokio::main]

async fn main() -> Result<(), Box> { // create client, reads STABILITYAPIKEY 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.