lucataco / ssd-lora-training

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

  • Public
  • 220 runs
  • GitHub

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