anotherjesse/ait-sd – Run with an API on Replicate

anotherjesse / ait-sd

Compile Stable Diffusion using AIT for A100

Cold

Public
24 runs
A100 (80GB)

Run with an API

Playground API Examples README Versions

Input

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

npx create-replicate --model=anotherjesse/ait-sd

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 anotherjesse/ait-sd using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "anotherjesse/ait-sd:66f7b1d6d72a3bcfb8a925d5d0608f6214e871b3aa370b12ddb446a0feba1d15",
  {
    input: {
      seed: 4896,
      width: 512,
      height: 512,
      num_outputs: 1,
      use_fp16_acc: true,
      convert_conv_to_gemm: true
    }
  }
);
console.log(output);

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 anotherjesse/ait-sd using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "anotherjesse/ait-sd:66f7b1d6d72a3bcfb8a925d5d0608f6214e871b3aa370b12ddb446a0feba1d15",
    input={
        "seed": 4896,
        "width": 512,
        "height": 512,
        "num_outputs": 1,
        "use_fp16_acc": True,
        "convert_conv_to_gemm": True
    }
)
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 anotherjesse/ait-sd 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": "66f7b1d6d72a3bcfb8a925d5d0608f6214e871b3aa370b12ddb446a0feba1d15",
    "input": {
      "seed": 4896,
      "width": 512,
      "height": 512,
      "num_outputs": 1,
      "use_fp16_acc": true,
      "convert_conv_to_gemm": true
    }
  }' \
  https://api.replicate.com/v1/predictions

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

You can run this model locally using Cog. First, install Cog:

brew install cog

If you don’t have Homebrew, there are other installation options available.

Run this to download the model and run it in your local environment:

cog predict r8.im/anotherjesse/ait-sd@sha256:66f7b1d6d72a3bcfb8a925d5d0608f6214e871b3aa370b12ddb446a0feba1d15 \
  -i 'seed=4896' \
  -i 'width=512' \
  -i 'height=512' \
  -i 'num_outputs=1' \
  -i 'use_fp16_acc=true' \
  -i 'convert_conv_to_gemm=true'

To learn more, take a look at the Cog documentation.

Run this to download the model and run it in your local environment:

docker run -d -p 5000:5000 --gpus=all r8.im/anotherjesse/ait-sd@sha256:66f7b1d6d72a3bcfb8a925d5d0608f6214e871b3aa370b12ddb446a0feba1d15
curl -s -X POST \
  -H "Content-Type: application/json" \
  -d $'{
    "input": {
      "seed": 4896,
      "width": 512,
      "height": 512,
      "num_outputs": 1,
      "use_fp16_acc": true,
      "convert_conv_to_gemm": true
    }
  }' \
  http://localhost:5000/predictions

To learn more, take a look at the Cog documentation.

Output

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

Run time and cost

This model runs on Nvidia A100 (80GB) GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

…

Model description

…

Intended use

…

Ethical considerations

…

Caveats and recommendations

…