swk23/ahsoka2 – Run with an API on Replicate

swk23 / ahsoka2

Warm

Public
90 runs
H100

Iterate in playground

Run with an API

Playground API Examples README Versions

Input

prompt

*string

Shift + Return to add a new line

Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives.Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives.

Prompt for generated image. If you include the `trigger_word` used in the training process you are more likely to activate the trained object, style, or concept in the resulting image.

image

file

Input image for image to image or inpainting mode. If provided, aspect_ratio, width, and height inputs are ignored.

mask

file

Image mask for image inpainting mode. If provided, aspect_ratio, width, and height inputs are ignored.

aspect_ratio

string

Aspect ratio for the generated image. If custom is selected, uses height and width below & will run in bf16 mode

Default: "1:1"

height

integer

(minimum: 256, maximum: 1440)

Height of generated image. Only works if `aspect_ratio` is set to custom. Will be rounded to nearest multiple of 16. Incompatible with fast generation

width

integer

(minimum: 256, maximum: 1440)

Width of generated image. Only works if `aspect_ratio` is set to custom. Will be rounded to nearest multiple of 16. Incompatible with fast generation

prompt_strength

number

(minimum: 0, maximum: 1)

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

Default: 0.8

model

string

Which model to run inference with. The dev model performs best with around 28 inference steps but the schnell model only needs 4 steps.

Default: "dev"

num_outputs

integer

(minimum: 1, maximum: 4)

Number of outputs to generate

Default: 1

num_inference_steps

integer

(minimum: 1, maximum: 50)

Number of denoising steps. More steps can give more detailed images, but take longer.

Default: 28

guidance_scale

number

(minimum: 0, maximum: 10)

Guidance scale for the diffusion process. Lower values can give more realistic images. Good values to try are 2, 2.5, 3 and 3.5

Default: 3

seed

integer

Random seed. Set for reproducible generation

output_format

string

Format of the output images

Default: "webp"

output_quality

integer

(minimum: 0, maximum: 100)

Quality when saving the output images, from 0 to 100. 100 is best quality, 0 is lowest quality. Not relevant for .png outputs

Default: 80

disable_safety_checker

boolean

Disable safety checker for generated images.

Default: false

go_fast

boolean

Run faster predictions with model optimized for speed (currently fp8 quantized); disable to run in original bf16

Default: false

megapixels

string

Approximate number of megapixels for generated image

Default: "1"

lora_scale

number

(minimum: -1, maximum: 3)

Determines how strongly the main LoRA should be applied. Sane results between 0 and 1 for base inference. For go_fast we apply a 1.5x multiplier to this value; we've generally seen good performance when scaling the base value by that amount. You may still need to experiment to find the best value for your particular lora.

Default: 1

extra_lora

string

Shift + Return to add a new line

Load LoRA weights. Supports Replicate models in the format <owner>/<username> or <owner>/<username>/<version>, HuggingFace URLs in the format huggingface.co/<owner>/<model-name>, CivitAI URLs in the format civitai.com/models/<id>[/<model-name>], or arbitrary .safetensors URLs from the Internet. For example, 'fofr/flux-pixar-cars'

extra_lora_scale

number

(minimum: -1, maximum: 3)

Determines how strongly the extra LoRA should be applied. Sane results between 0 and 1 for base inference. For go_fast we apply a 1.5x multiplier to this value; we've generally seen good performance when scaling the base value by that amount. You may still need to experiment to find the best value for your particular lora.

Default: 1

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

npx create-replicate --model=swk23/ahsoka2

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

const output = await replicate.run(
  "swk23/ahsoka2:56e053a5386c69a6e4914392cda9d3e4b2d393db23283705b7943973c581563d",
  {
    input: {
      model: "dev",
      prompt: "Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives.",
      go_fast: false,
      lora_scale: 1,
      megapixels: "1",
      num_outputs: 1,
      aspect_ratio: "21:9",
      output_format: "jpg",
      guidance_scale: 3,
      output_quality: 80,
      prompt_strength: 0.8,
      extra_lora_scale: 1,
      num_inference_steps: 28
    }
  }
);

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

output = replicate.run(
    "swk23/ahsoka2:56e053a5386c69a6e4914392cda9d3e4b2d393db23283705b7943973c581563d",
    input={
        "model": "dev",
        "prompt": "Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives.",
        "go_fast": False,
        "lora_scale": 1,
        "megapixels": "1",
        "num_outputs": 1,
        "aspect_ratio": "21:9",
        "output_format": "jpg",
        "guidance_scale": 3,
        "output_quality": 80,
        "prompt_strength": 0.8,
        "extra_lora_scale": 1,
        "num_inference_steps": 28
    }
)
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 swk23/ahsoka2 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": "swk23/ahsoka2:56e053a5386c69a6e4914392cda9d3e4b2d393db23283705b7943973c581563d",
    "input": {
      "model": "dev",
      "prompt": "Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives.",
      "go_fast": false,
      "lora_scale": 1,
      "megapixels": "1",
      "num_outputs": 1,
      "aspect_ratio": "21:9",
      "output_format": "jpg",
      "guidance_scale": 3,
      "output_quality": 80,
      "prompt_strength": 0.8,
      "extra_lora_scale": 1,
      "num_inference_steps": 28
    }
  }' \
  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/swk23/ahsoka2@sha256:56e053a5386c69a6e4914392cda9d3e4b2d393db23283705b7943973c581563d \
  -i 'model="dev"' \
  -i 'prompt="Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives."' \
  -i 'go_fast=false' \
  -i 'lora_scale=1' \
  -i 'megapixels="1"' \
  -i 'num_outputs=1' \
  -i 'aspect_ratio="21:9"' \
  -i 'output_format="jpg"' \
  -i 'guidance_scale=3' \
  -i 'output_quality=80' \
  -i 'prompt_strength=0.8' \
  -i 'extra_lora_scale=1' \
  -i 'num_inference_steps=28'

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/swk23/ahsoka2@sha256:56e053a5386c69a6e4914392cda9d3e4b2d393db23283705b7943973c581563d
curl -s -X POST \
  -H "Content-Type: application/json" \
  -d $'{
    "input": {
      "model": "dev",
      "prompt": "Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives.",
      "go_fast": false,
      "lora_scale": 1,
      "megapixels": "1",
      "num_outputs": 1,
      "aspect_ratio": "21:9",
      "output_format": "jpg",
      "guidance_scale": 3,
      "output_quality": 80,
      "prompt_strength": 0.8,
      "extra_lora_scale": 1,
      "num_inference_steps": 28
    }
  }' \
  http://localhost:5000/predictions

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

Output

{
  "completed_at": "2025-01-14T23:52:39.508759Z",
  "created_at": "2025-01-14T23:52:32.288000Z",
  "data_removed": false,
  "error": null,
  "id": "zyp4gcxf41rm80cmcwxt8gy930",
  "input": {
    "model": "dev",
    "prompt": "Ahsoka Tano working alongside farmers in a rustic settlement, reinforcing a damaged fence. Her twin lekku sway as she uses simple tools, her movements precise and steady. The farmers, dressed in rugged, practical clothing, assist her with determination, their hands worn from hard labor. In the background, fields stretch under a warm sunset, and simple farming machines rest near the crops. The air is filled with dust and the hum of distant activity, as Ahsoka’s calm and focused expression reflects her commitment to helping rebuild their lives.",
    "go_fast": false,
    "lora_scale": 1,
    "megapixels": "1",
    "num_outputs": 1,
    "aspect_ratio": "21:9",
    "output_format": "jpg",
    "guidance_scale": 3,
    "output_quality": 80,
    "prompt_strength": 0.8,
    "extra_lora_scale": 1,
    "num_inference_steps": 28
  },
  "logs": "2025-01-14 23:52:32.499 | DEBUG    | fp8.lora_loading:apply_lora_to_model:574 - Extracting keys\n2025-01-14 23:52:32.499 | DEBUG    | fp8.lora_loading:apply_lora_to_model:581 - Keys extracted\nApplying LoRA:   0%|          | 0/304 [00:00<?, ?it/s]\nApplying LoRA:  93%|█████████▎| 283/304 [00:00<00:00, 2822.41it/s]\nApplying LoRA: 100%|██████████| 304/304 [00:00<00:00, 2726.28it/s]\n2025-01-14 23:52:32.611 | SUCCESS  | fp8.lora_loading:unload_loras:564 - LoRAs unloaded in 0.11s\nfree=29172608561152\nDownloading weights\n2025-01-14T23:52:32Z | INFO  | [ Initiating ] chunk_size=150M dest=/tmp/tmpj_qysa6g/weights url=https://replicate.delivery/xezq/S7yoWLIFwUI7C1n4ne2aZ1oXn2QYUtXp9RWqQ4kH25h7Uz9JA/trained_model.tar\n2025-01-14T23:52:33Z | INFO  | [ Complete ] dest=/tmp/tmpj_qysa6g/weights size=\"172 MB\" total_elapsed=1.085s url=https://replicate.delivery/xezq/S7yoWLIFwUI7C1n4ne2aZ1oXn2QYUtXp9RWqQ4kH25h7Uz9JA/trained_model.tar\nDownloaded weights in 1.16s\n2025-01-14 23:52:33.774 | INFO     | fp8.lora_loading:convert_lora_weights:498 - Loading LoRA weights for /src/weights-cache/fc1e8437fea32199\n2025-01-14 23:52:33.843 | INFO     | fp8.lora_loading:convert_lora_weights:519 - LoRA weights loaded\n2025-01-14 23:52:33.843 | DEBUG    | fp8.lora_loading:apply_lora_to_model:574 - Extracting keys\n2025-01-14 23:52:33.843 | DEBUG    | fp8.lora_loading:apply_lora_to_model:581 - Keys extracted\nApplying LoRA:   0%|          | 0/304 [00:00<?, ?it/s]\nApplying LoRA:  93%|█████████▎| 283/304 [00:00<00:00, 2828.20it/s]\nApplying LoRA: 100%|██████████| 304/304 [00:00<00:00, 2731.44it/s]\n2025-01-14 23:52:33.955 | SUCCESS  | fp8.lora_loading:load_lora:539 - LoRA applied in 0.18s\nUsing seed: 42997\n0it [00:00, ?it/s]\n1it [00:00,  9.12it/s]\n2it [00:00,  6.37it/s]\n3it [00:00,  5.82it/s]\n4it [00:00,  5.60it/s]\n5it [00:00,  5.48it/s]\n6it [00:01,  5.37it/s]\n7it [00:01,  5.33it/s]\n8it [00:01,  5.32it/s]\n9it [00:01,  5.31it/s]\n10it [00:01,  5.29it/s]\n11it [00:02,  5.26it/s]\n12it [00:02,  5.26it/s]\n13it [00:02,  5.24it/s]\n14it [00:02,  5.22it/s]\n15it [00:02,  5.23it/s]\n16it [00:02,  5.21it/s]\n17it [00:03,  5.24it/s]\n18it [00:03,  5.23it/s]\n19it [00:03,  5.25it/s]\n20it [00:03,  5.23it/s]\n21it [00:03,  5.24it/s]\n22it [00:04,  5.22it/s]\n23it [00:04,  5.22it/s]\n24it [00:04,  5.24it/s]\n25it [00:04,  5.23it/s]\n26it [00:04,  5.23it/s]\n27it [00:05,  5.23it/s]\n28it [00:05,  5.21it/s]\n28it [00:05,  5.32it/s]\nTotal safe images: 1 out of 1",
  "metrics": {
    "predict_time": 7.008653102,
    "total_time": 7.220759
  },
  "output": [
    "https://replicate.delivery/xezq/SZkYwnP6jmpNBVUCSp7AXkQraSqKcemzpx3WeAUMeCYOWVKoA/out-0.jpg"
  ],
  "started_at": "2025-01-14T23:52:32.500106Z",
  "status": "succeeded",
  "urls": {
    "stream": "https://stream.replicate.com/v1/files/bcwr-izm4ar6vx24hth5spwl6rdmhg2vuzelkyp4gn42ih5rkyat6leva",
    "get": "https://api.replicate.com/v1/predictions/zyp4gcxf41rm80cmcwxt8gy930",
    "cancel": "https://api.replicate.com/v1/predictions/zyp4gcxf41rm80cmcwxt8gy930/cancel"
  },
  "version": "56e053a5386c69a6e4914392cda9d3e4b2d393db23283705b7943973c581563d"
}

Generated in

7.0 seconds

Tweak it ShareReport

2025-01-14 23:52:32.499 | DEBUG    | fp8.lora_loading:apply_lora_to_model:574 - Extracting keys
2025-01-14 23:52:32.499 | DEBUG    | fp8.lora_loading:apply_lora_to_model:581 - Keys extracted
Applying LoRA:   0%|          | 0/304 [00:00<?, ?it/s]
Applying LoRA:  93%|█████████▎| 283/304 [00:00<00:00, 2822.41it/s]
Applying LoRA: 100%|██████████| 304/304 [00:00<00:00, 2726.28it/s]
2025-01-14 23:52:32.611 | SUCCESS  | fp8.lora_loading:unload_loras:564 - LoRAs unloaded in 0.11s
free=29172608561152
Downloading weights
2025-01-14T23:52:32Z | INFO  | [ Initiating ] chunk_size=150M dest=/tmp/tmpj_qysa6g/weights url=https://replicate.delivery/xezq/S7yoWLIFwUI7C1n4ne2aZ1oXn2QYUtXp9RWqQ4kH25h7Uz9JA/trained_model.tar
2025-01-14T23:52:33Z | INFO  | [ Complete ] dest=/tmp/tmpj_qysa6g/weights size="172 MB" total_elapsed=1.085s url=https://replicate.delivery/xezq/S7yoWLIFwUI7C1n4ne2aZ1oXn2QYUtXp9RWqQ4kH25h7Uz9JA/trained_model.tar
Downloaded weights in 1.16s
2025-01-14 23:52:33.774 | INFO     | fp8.lora_loading:convert_lora_weights:498 - Loading LoRA weights for /src/weights-cache/fc1e8437fea32199
2025-01-14 23:52:33.843 | INFO     | fp8.lora_loading:convert_lora_weights:519 - LoRA weights loaded
2025-01-14 23:52:33.843 | DEBUG    | fp8.lora_loading:apply_lora_to_model:574 - Extracting keys
2025-01-14 23:52:33.843 | DEBUG    | fp8.lora_loading:apply_lora_to_model:581 - Keys extracted
Applying LoRA:   0%|          | 0/304 [00:00<?, ?it/s]
Applying LoRA:  93%|█████████▎| 283/304 [00:00<00:00, 2828.20it/s]
Applying LoRA: 100%|██████████| 304/304 [00:00<00:00, 2731.44it/s]
2025-01-14 23:52:33.955 | SUCCESS  | fp8.lora_loading:load_lora:539 - LoRA applied in 0.18s
Using seed: 42997
0it [00:00, ?it/s]
1it [00:00,  9.12it/s]
2it [00:00,  6.37it/s]
3it [00:00,  5.82it/s]
4it [00:00,  5.60it/s]
5it [00:00,  5.48it/s]
6it [00:01,  5.37it/s]
7it [00:01,  5.33it/s]
8it [00:01,  5.32it/s]
9it [00:01,  5.31it/s]
10it [00:01,  5.29it/s]
11it [00:02,  5.26it/s]
12it [00:02,  5.26it/s]
13it [00:02,  5.24it/s]
14it [00:02,  5.22it/s]
15it [00:02,  5.23it/s]
16it [00:02,  5.21it/s]
17it [00:03,  5.24it/s]
18it [00:03,  5.23it/s]
19it [00:03,  5.25it/s]
20it [00:03,  5.23it/s]
21it [00:03,  5.24it/s]
22it [00:04,  5.22it/s]
23it [00:04,  5.22it/s]
24it [00:04,  5.24it/s]
25it [00:04,  5.23it/s]
26it [00:04,  5.23it/s]
27it [00:05,  5.23it/s]
28it [00:05,  5.21it/s]
28it [00:05,  5.32it/s]
Total safe images: 1 out of 1