Wiro - Train Dreambooth Sd icon

Wiro - Train Dreambooth Sd

Train your own custom Stable Diffusion model using a small set of images

Wiro - Train Dreambooth Sd

AI-generated

Overview

This node allows users to train a custom Stable Diffusion model using a small set of images by interacting with the Wiro API. It is useful for scenarios where users want to create personalized AI models for image generation based on their own datasets and training parameters. For example, artists or developers can train models with specific prompts and datasets to generate unique images tailored to their needs.

Use Case Examples

  1. Training a custom Stable Diffusion model with a specific dataset and prompts to generate personalized artwork.
  2. Using the node to automate the training process of AI models with configurable parameters like learning rate, batch size, and training steps.

Properties

Name Meaning
Trained Model Name The name of the model to be trained, required to identify the training job.
Train Model Description Optional description of the model being trained.
Reference Model The base model to use as a reference for training.
Reference Model Private Indicates if the reference model is private.
Classification Dataset The dataset used for classification during training, required.
Instance Prompt Prompt describing the instance for training, required.
Class Prompt Prompt describing the class for training, required.
Save Sample Prompt Prompt used for saving sample images.
Save Sample Negative Prompt Negative prompt used for saving sample images.
Class Images Number of class images used in training, required.
N Save Sample Number of samples to save, required.
Save Guidance Scale Guidance scale used when saving samples, required.
Save Infer Steps Number of inference steps used when saving samples, required.
Pad Tokens Boolean flag to enable padding tokens.
With Prior Preservation Boolean flag to enable prior preservation during training.
Prior Loss Weight Weight of the prior loss during training, required.
Seed Seed value for reproducibility, required.
Resolution Resolution of the training images, required.
Center Crop Boolean flag to enable center cropping of images.
Train Text Encoder Boolean flag to enable training of the text encoder.
Train Batch Size Batch size used during training, required.
Sample Batch Size Batch size used during sampling, required.
Training Steps Number of training steps, required.
Gradient Accumulation Steps Number of gradient accumulation steps, required.
Gradient Checkpointing Boolean flag to enable gradient checkpointing.
Learning Rate Learning rate used during training, required.
Scale Lr Boolean flag to enable learning rate scaling.
Lr Scheduler Learning rate scheduler type, required.
Lr Warm Up Steps Number of warm-up steps for learning rate, required.
Use 8bit Adam Boolean flag to enable 8-bit Adam optimizer.
Adam Beta1 Beta1 parameter for Adam optimizer, required.
Adam Beta2 Beta2 parameter for Adam optimizer, required.
Adam Weight Decay Weight decay parameter for Adam optimizer, required.
Adam Epsilon Epsilon parameter for Adam optimizer, required.
Max Grad Norm Maximum gradient norm for clipping, required.
Save Interval Interval for saving the model during training, required.
Save Min Steps Minimum steps before saving the model, required.
Mixed Precision Mixed precision mode used during training, required.
Selected Folder For Train Data Folder path for training data.
ZIP URL For Training Data URL to a ZIP file containing training data, which will be extracted automatically.

Output

JSON

  • taskid - The unique identifier of the training task.
  • url - URL to access the trained model or training results.
  • status - Status of the training task, e.g., 'completed' or 'failed'.

Dependencies

  • Wiro API with API key and secret for authentication

Troubleshooting

  • If the API response does not contain a valid task ID or socket access token, the node throws an error indicating an invalid response from the Wiro API. Ensure that the API credentials are correct and the Wiro service is available.
  • Network or authentication issues may cause the HTTP request to fail. Verify network connectivity and API key/secret validity.
  • If the polling for task completion returns error codes (-1, -2, -3, -4), the task status is marked as 'failed'. Check the Wiro API documentation for error code meanings and adjust input parameters accordingly.

Links

  • Wiro API Documentation - Official documentation for the Wiro API used for training Dreambooth Stable Diffusion models.

Discussion