zeke/zekebooth-cog-stable-diffusion – Run with an API on Replicate

zeke / zekebooth-cog-stable-diffusion

Cold

Public
56 runs
A100 (80GB)

Iterate in playground

Run with an API

Playground API Examples README Versions

Input

prompt

string

Shift + Return to add a new line

a photo of an astronaut riding a horse on marsa photo of an astronaut riding a horse on mars

Input prompt

Default: "a photo of an astronaut riding a horse on mars"

negative_prompt

string

Shift + Return to add a new line

Specify things to not see in the output

width

integer

Width of output image. Maximum size is 1024x768 or 768x1024 because of memory limits

Default: 768

height

integer

Height of output image. Maximum size is 1024x768 or 768x1024 because of memory limits

Default: 768

prompt_strength

number

Prompt strength when using init image. 1.0 corresponds to full destruction of information in init image

Default: 0.8

num_outputs

integer

(minimum: 1, maximum: 4)

Number of images to output.

Default: 1

num_inference_steps

integer

(minimum: 1, maximum: 500)

Number of denoising steps

Default: 50

guidance_scale

number

(minimum: 1, maximum: 20)

Scale for classifier-free guidance

Default: 7.5

scheduler

string

Choose a scheduler

Default: "K_EULER"

seed

integer

Random seed. Leave blank to randomize the seed

Run this model in Node.js with one line of code:

npx create-replicate --model=zeke/zekebooth-cog-stable-diffusion

or set up a project from scratch

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import and set up the client:

import Replicate from "replicate";

const replicate = new Replicate({
  auth: process.env.REPLICATE_API_TOKEN,
});

Run zeke/zekebooth-cog-stable-diffusion using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "zeke/zekebooth-cog-stable-diffusion:190bb4b6100291f26ee4fc8939daf9362984329c828561b85f9af690ac63e13c",
  {
    input: {
      width: 768,
      height: 768,
      prompt: "a photo of an astronaut riding a horse on mars",
      scheduler: "K_EULER",
      num_outputs: 1,
      guidance_scale: 7.5,
      prompt_strength: 0.8,
      num_inference_steps: 50
    }
  }
);

// To access the file URL:
console.log(output[0].url()); //=> "http://example.com"

// To write the file to disk:
fs.writeFile("my-image.png", output[0]);

To learn more, take a look at the guide on getting started with Node.js.

Install Replicate’s Python client library:

pip install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import the client:

import replicate

Run zeke/zekebooth-cog-stable-diffusion using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "zeke/zekebooth-cog-stable-diffusion:190bb4b6100291f26ee4fc8939daf9362984329c828561b85f9af690ac63e13c",
    input={
        "width": 768,
        "height": 768,
        "prompt": "a photo of an astronaut riding a horse on mars",
        "scheduler": "K_EULER",
        "num_outputs": 1,
        "guidance_scale": 7.5,
        "prompt_strength": 0.8,
        "num_inference_steps": 50
    }
)
print(output)

To learn more, take a look at the guide on getting started with Python.

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run zeke/zekebooth-cog-stable-diffusion using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

curl -s -X POST \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -H "Prefer: wait" \
  -d $'{
    "version": "zeke/zekebooth-cog-stable-diffusion:190bb4b6100291f26ee4fc8939daf9362984329c828561b85f9af690ac63e13c",
    "input": {
      "width": 768,
      "height": 768,
      "prompt": "a photo of an astronaut riding a horse on mars",
      "scheduler": "K_EULER",
      "num_outputs": 1,
      "guidance_scale": 7.5,
      "prompt_strength": 0.8,
      "num_inference_steps": 50
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

Output

No output yet! Press "Submit" to start a prediction.

Run time and cost

This model costs approximately $0.0086 to run on Replicate, or 116 runs per $1, but this varies depending on your inputs. It is also open source and you can run it on your own computer with Docker.

This model runs on Nvidia A100 (80GB) GPU hardware. Predictions typically complete within 7 seconds.

Readme

…

Model description

…

Intended use

…

Ethical considerations

…

Caveats and recommendations

…