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

Add Full LoRa Training

Choose a tag to compare

View all tags
@ExponentialML ExponentialML released this 11 Apr 02:05
v2.2.0
053b517
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

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.
Assets 2
Loading