Jaret Burkett 27f343fc08 Added base setup for training t2i adapters. Currently untested, saw something else shiny i wanted to finish sirst. Added content_or_style to the training config. It defaults to balanced, which is standard uniform time step sampling. If style or content is passed, it will use cubic sampling for timesteps to favor timesteps that are beneficial for training them. for style, favor later timesteps. For content, favor earlier timesteps.

2023-09-16 08:30:38 -06:00

..

config

Hude rework to move the batch to a DTO to make it far more modular to the future ui

2023-08-29 10:22:19 -06:00

__init__.py

Added support for training lora, dreambooth, and fine tuning. Still need testing and docs

2023-08-23 15:37:00 -06:00

SDTrainer.py

Added base setup for training t2i adapters. Currently untested, saw something else shiny i wanted to finish sirst. Added content_or_style to the training config. It defaults to balanced, which is standard uniform time step sampling. If style or content is passed, it will use cubic sampling for timesteps to favor timesteps that are beneficial for training them. for style, favor later timesteps. For content, favor earlier timesteps.

2023-09-16 08:30:38 -06:00