prompthunt / traintest

(Updated 1 year, 7 months ago)

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

output = replicate.run(
    "prompthunt/traintest:e32d635c13d645cbdd420b799522256f468f58f2a8fc2aeca9d7097709f20ecd",
    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": 2560,
        "num_train_epochs": 4000,
        "train_batch_size": 1,
        "unet_learning_rate": 0.000001,
        "checkpointing_steps": 999999,
        "clipseg_temperature": 1,
        "input_images_filetype": "infer",
        "crop_based_on_salience": False,
        "use_face_detection_instead": True
    }
)
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.