prompthunt / cog-sdxl-training

  • Public
  • 7 runs
  • L40S
Iterate in playground

Input

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

output = replicate.run(
    "prompthunt/cog-sdxl-training:bef5d92c784ae23bb8c5f79ceb4e58fb581c039513be7c4e2d60de349f132b6e",
    input={
        "ti_lr": 0.0003,
        "is_lora": True,
        "lora_lr": 0.0001,
        "verbose": True,
        "lora_rank": 32,
        "resolution": 768,
        "lr_scheduler": "constant",
        "token_string": "TOK",
        "caption_prefix": "a photo of TOK, ",
        "lr_warmup_steps": 100,
        "max_train_steps": 1000,
        "num_train_epochs": 4000,
        "train_batch_size": 4,
        "unet_learning_rate": 0.000001,
        "checkpointing_steps": 999999,
        "clipseg_temperature": 1,
        "input_images_filetype": "infer",
        "crop_based_on_salience": True,
        "use_face_detection_instead": False
    }
)
print(output)

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

Output

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

Run time and cost

This model runs on Nvidia L40S GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

This model doesn't have a readme.