lucataco / ssd-lora-training

POC to train SSD-1B LoRAs for cheaper & faster training

  • Public
  • 221 runs
  • GitHub
Run with an API
Playground API Examples README Versions

Input

Output

Run time and cost

This model costs approximately $0.20 to run on Replicate, or 5 runs per $1, but this varies depending on your inputs. It is also open source and you can run it on your own computer with Docker.

This model runs on Nvidia A40 (Large) GPU hardware. Predictions typically complete within 5 minutes. The predict time for this model varies significantly based on the inputs.

Readme

About

This is a hacked together, proof of concept model to train your own SSD-1B LoRAs. At this time, it does not support the standard Replicate training method (replicate.trainings.create).

I took the original SDXL parameters and just halved the num_train_epochs, and max_train_steps

Goal: To create SSD LoRAs at half the time, and half the cost of SDXL, but with no loss in quality

All you need are your training images in a zip file and select “use_face_detection_instead” if training faces

Run inference on SSD-1B LoRAs here