lucataco / ssd-lora-training

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

  • Public
  • 221 runs
  • GitHub

Input

Output

Run time and cost

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