ai-toolkit/extensions_built_in/sd_trainer
Jaret Burkett 4fa8fac5fd WIP multidevice training 2024-08-29 16:04:20 -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
SDTrainer.py Made peleminary arch for flux ip adapter training 2024-08-28 08:55:39 -06:00
TrainerV2.py WIP multidevice training 2024-08-29 16:04:20 -06:00
__init__.py Created a v2 trainer and moved all the training logic to single torch model so it can can be run in parallel 2024-08-29 12:34:18 -06:00