ComfyUI/comfy/ldm/flux
Kohaku-Blueleaf cdcf4119b3
[Trainer] training with proper offloading (#12189)
* Fix bypass dtype/device moving

* Force offloading mode for training

* training context var

* offloading implementation in training node

* fix wrong input type

* Support bypass load lora model, correct adapter/offloading handling
2026-02-10 21:45:19 -05:00
..
controlnet.py Make flux controlnet work with sd3 text enc. (#8599) 2025-06-19 18:50:05 -04:00
layers.py Implement the Ovis image model. (#11030) 2025-12-01 20:56:17 -05:00
math.py [Trainer] training with proper offloading (#12189) 2026-02-10 21:45:19 -05:00
model.py Implement the Ovis image model. (#11030) 2025-12-01 20:56:17 -05:00
redux.py Support new flux model variants. 2024-11-21 08:38:23 -05:00