Failed to load versions. Head to the versions page to see all versions for this model.
You're looking at a specific version of this model. Jump to the model overview.
lucataco /train-text-to-image-lora:aa769732
Input
Run this model in Node.js with one line of code:
npm install replicate
REPLICATE_API_TOKEN
environment variableexport REPLICATE_API_TOKEN=<paste-your-token-here>
Find your API token in your account settings.
import Replicate from "replicate";
const replicate = new Replicate({
auth: process.env.REPLICATE_API_TOKEN,
});
Run lucataco/train-text-to-image-lora using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.
const output = await replicate.run(
"lucataco/train-text-to-image-lora:aa769732ccfc922fb77ec36c1bd3680208394f1770402db52681fc8556e9356a",
{
input: {
dataset: "lambdalabs/naruto-blip-captions",
base_model: "runwayml/stable-diffusion-v1-5",
resolution: 512,
hub_model_id: "naruto-lora",
lr_scheduler: "cosine",
learning_rate: 0.0001,
max_grad_norm: 1,
max_train_steps: 1000,
num_train_epochs: 100,
train_batch_size: 1,
validation_prompt: "A naruto with blue eyes.",
dataloader_num_workers: 8,
gradient_accumulation_steps: 4
}
}
);
console.log(output);
To learn more, take a look at the guide on getting started with Node.js.
pip install replicate
REPLICATE_API_TOKEN
environment variableexport REPLICATE_API_TOKEN=<paste-your-token-here>
Find your API token in your account settings.
import replicate
Run lucataco/train-text-to-image-lora using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.
output = replicate.run(
"lucataco/train-text-to-image-lora:aa769732ccfc922fb77ec36c1bd3680208394f1770402db52681fc8556e9356a",
input={
"dataset": "lambdalabs/naruto-blip-captions",
"base_model": "runwayml/stable-diffusion-v1-5",
"resolution": 512,
"hub_model_id": "naruto-lora",
"lr_scheduler": "cosine",
"learning_rate": 0.0001,
"max_grad_norm": 1,
"max_train_steps": 1000,
"num_train_epochs": 100,
"train_batch_size": 1,
"validation_prompt": "A naruto with blue eyes.",
"dataloader_num_workers": 8,
"gradient_accumulation_steps": 4
}
)
print(output)
To learn more, take a look at the guide on getting started with Python.
REPLICATE_API_TOKEN
environment variableexport REPLICATE_API_TOKEN=<paste-your-token-here>
Find your API token in your account settings.
Run lucataco/train-text-to-image-lora 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": "aa769732ccfc922fb77ec36c1bd3680208394f1770402db52681fc8556e9356a",
"input": {
"dataset": "lambdalabs/naruto-blip-captions",
"base_model": "runwayml/stable-diffusion-v1-5",
"resolution": 512,
"hub_model_id": "naruto-lora",
"lr_scheduler": "cosine",
"learning_rate": 0.0001,
"max_grad_norm": 1,
"max_train_steps": 1000,
"num_train_epochs": 100,
"train_batch_size": 1,
"validation_prompt": "A naruto with blue eyes.",
"dataloader_num_workers": 8,
"gradient_accumulation_steps": 4
}
}' \
https://api.replicate.com/v1/predictions
To learn more, take a look at Replicate’s HTTP API reference docs.
Add a payment method to run this model.
By signing in, you agree to our
terms of service and privacy policy
Output
No output yet! Press "Submit" to start a prediction.