Kohaku-Blueleaf
12aacb978b
Fix bad id/naming
2025-11-11 16:37:27 +08:00
Kohaku-Blueleaf
1f852778ec
Add outputs' id/name
2025-11-11 16:23:24 +08:00
Kohaku-Blueleaf
b85ed81792
add ui pbar
2025-11-08 15:42:18 +08:00
Kohaku-Blueleaf
b3784a7da1
Rewrite training system with new io schema
2025-11-08 15:06:48 +08:00
Kohaku-Blueleaf
650b7b0302
move all dataset related implementation to nodes_dataset
2025-11-04 12:51:40 +08:00
Kohaku-Blueleaf
992aa2dd8f
allow trainer node to get different resolution dataset
2025-11-04 12:50:25 +08:00
Kohaku-Blueleaf
32b44f5d1c
make training node to work with our dataset system
2025-11-03 00:33:54 +08:00
Kohaku-Blueleaf
7be2b49b6b
Fix LoRA Trainer bugs with FP8 models. ( #9854 )
...
* Fix adapter weight init
* Fix fp8 model training
* Avoid inference tensor
2025-09-20 21:24:48 -04:00
Kohaku-Blueleaf
eb2f78b4e0
[Training Node] algo support, grad acc, optional grad ckpt ( #9015 )
...
* Add factorization utils for lokr
* Add lokr train impl
* Add loha train impl
* Add adapter map for algo selection
* Add optional grad ckpt and algo selection
* Update __init__.py
* correct key name for loha
* Use custom fwd/bwd func and better init for loha
* Support gradient accumulation
* Fix bugs of loha
* use more stable init
* Add OFT training
* linting
2025-07-23 20:57:27 -04:00
Kohaku-Blueleaf
1205afc708
Better training loop implementation ( #8820 )
2025-07-09 11:41:22 -04:00
Kohaku-Blueleaf
181a9bf26d
Support Multi Image-Caption dataset in lora training node ( #8819 )
...
* initial impl of multi img/text dataset
* Update nodes_train.py
* Support Kohya-ss structure
2025-07-08 20:18:04 -04:00
comfyanonymous
ae75a084df
SaveLora now saves in the same filename format as all the other nodes. ( #8538 )
2025-06-15 03:44:59 -04:00
Kohaku-Blueleaf
520eb77b72
LoRA Trainer: LoRA training node in weight adapter scheme ( #8446 )
2025-06-13 19:25:59 -04:00
