ComfyUI Node
Flux2TTRControllerTrainer
Phase-2: train a TTR controller using dual-path sampling and REINFORCE updates.
Flux2TTRControllerTrainer
- model_original
- model_ttr
- positive
- negative
- latent
- vae
- training_config
- IMAGE_ORIGINAL
- IMAGE_PATCHED
- CONTROLLER
◄steps20►
◄seed0►
◄cfg1.000►
◄sampler_nameeuler►
◄schedulernormal►
◄checkpoint_path/tmp/ComfyUI/models/approximate_attention/flux2_ttr_controller.pt►
◄training_iterations100►
◄denoise1.000►
◄sigma_aware_trainingtrue►
Categoryadvanced/attention
Inputs (16)
| Name | Type | Default | Description |
|---|---|---|---|
| model_original | MODEL | — | |
| model_ttr | MODEL | — | |
| positive | CONDITIONING | — | |
| negative | CONDITIONING | — | |
| latent | LATENT | — | |
| vae | VAE | — | |
| training_config | TTR_TRAINING_CONFIG | — | |
| steps | INT | 201–10000 | — |
| seed | INT | 00–18446744073709550000 | — |
| cfg | FLOAT | 1.0000–100 | — |
| sampler_name | COMBO | euler | 44 options: euler, euler_cfg_pp, euler_ancestral, euler_ancestral_cfg_pp, heun, heunpp2, +38 |
| scheduler | COMBO | normal | 9 options: simple, sgm_uniform, karras, exponential, ddim_uniform, beta, +3 |
| checkpoint_path | STRING | /tmp/ComfyUI/models/approximate_attention/flux2_ttr_controller.pt | Controller checkpoint path (defaults to ComfyUI/models/approximate_attention/flux2_ttr_controller.pt). |
| training_iterations | INT | 1001–100000 | — |
| denoise | FLOAT | 1.0000–1 | — |
| sigma_aware_training | BOOLEAN | true | — |
Outputs (3)
| Name | Type | Description |
|---|---|---|
| IMAGE_ORIGINAL | IMAGE | — |
| IMAGE_PATCHED | IMAGE | — |
| CONTROLLER | TTR_CONTROLLER | — |