Nodes/ComfyUI-Taylor-Attention/Flux2TTRTrainingParameters
ComfyUI Node

Flux2TTRTrainingParameters

Shared Flux2TTR training hyperparameters for Phase-1 and Phase-2 nodes.

By ttulttul·Created 6 months ago·Updated 6 months ago· 1
Flux2TTRTrainingParameters
    • training_config
    rmse_weight1.000
    cosine_weight1.000
    lpips_weight0.000
    dreamsim_weight0.000
    hps_weight0.000
    biqa_quality_weight0.000
    biqa_aesthetic_weight0.000
    gemini_similarity_weight0.000
    gemini_quality_weight0.000
    gemini_modelgemini-3-flash-preview
    gemini_api_key
    gemini_api_key_envGEMINI_API_KEY
    reward_baseline_quality_floor-0.300
    huber_beta0.0500
    learning_rate0
    grad_clip_norm1.000
    alpha_lr_multiplier5.000
    phi_lr_multiplier1.000
    target_ttr_ratio0.700
    lambda_eff1.000
    lambda_entropy0.100
    gumbel_temperature_start1.000
    gumbel_temperature_end0.500
    warmup_steps0
    comet_enabledfalse
    comet_api_key
    comet_project_namettr-distillation
    comet_workspace
    comet_experimentc6376d52026072021405003308600636
    log_every10
    Categoryadvanced/attention

    Inputs (30)

    NameTypeDefaultDescription
    rmse_weightFLOAT1.0000–1000
    cosine_weightFLOAT1.0000–1000
    lpips_weightFLOAT0.0000–1000
    dreamsim_weightFLOAT0.0000–1000
    hps_weightFLOAT0.0000–1000
    biqa_quality_weightFLOAT0.0000–1000
    biqa_aesthetic_weightFLOAT0.0000–1000
    gemini_similarity_weightFLOAT0.0000–1000
    gemini_quality_weightFLOAT0.0000–1000
    gemini_modelSTRINGgemini-3-flash-previewGemini model used for image-to-image quality scoring when Gemini weights are enabled.
    gemini_api_keySTRINGOptional Gemini API key value. If empty, gemini_api_key_env is used.
    gemini_api_key_envSTRINGGEMINI_API_KEYEnvironment variable name that holds the Gemini API key.
    reward_baseline_quality_floorFLOAT-0.300-10–1Clamp floor for reward-baseline EMA updates in controller training.
    huber_betaFLOAT0.05000.000001–10
    learning_rateFLOAT01e-7–1
    grad_clip_normFLOAT1.0000–100
    alpha_lr_multiplierFLOAT5.0000–100
    phi_lr_multiplierFLOAT1.0000–100
    target_ttr_ratioFLOAT0.7000–1
    lambda_effFLOAT1.0000–100
    lambda_entropyFLOAT0.1000–1
    gumbel_temperature_startFLOAT1.0000.0001–10
    gumbel_temperature_endFLOAT0.5000.0001–10
    warmup_stepsINT00–1000000
    comet_enabledBOOLEANfalse
    comet_api_keySTRING
    comet_project_nameSTRINGttr-distillation
    comet_workspaceSTRING
    comet_experimentSTRINGc6376d52026072021405003308600636Comet experiment key. Use a stable value to keep logging to one persistent experiment.
    log_everyINT101–1000000

    Outputs (1)

    NameTypeDescription
    training_configTTR_TRAINING_CONFIG