Skip to content
This repository has been archived by the owner on Dec 14, 2023. It is now read-only.

Add Full LoRa Training

Compare
Choose a tag to compare
@ExponentialML ExponentialML released this 11 Apr 02:05
· 191 commits to main since this release
053b517

What's New

  • LoRa training based off of cloneofsimo's repository.
  • Add LoraInjectedConv3d module. 🎥
  • Add config for LoRA only training.
  • Add option to save LoRA for UNet & Text Encoder.
  • Fix checkpointing model files during training.