edenartlab / sdxl-lora-trainer

LoRa trainer for both SDXL and SD15

  • Public
  • 5.1K runs
  • GitHub
  • License
Run with an API
Playground API Examples README Versions

Input

Output

Run time and cost

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

Readme

This trainer uses a single training script that is compatible with both SDXL and SD15.

The trainer has the following capabilities: - automatic image captioning using BLIP - automatic segmentation using CLIPseg - textual_inversion training of a new token to represent the concept - 3 training modes: “style” / “face” / “object” - Full finetuning or LoRa or Dora training modes are supported in the code - LoRa modules are possible for both unet and txt-encoders

The generated checkpoint files are compatible with ComfyUI and AUTO111