flux2_modulation

FLUX.2 shared modulation projection

Description

linear(silu(temb)) producing mod_param_sets triples of (shift, scale, gate). Computed once per forward at model level and broadcast to every block. Reference: Flux2Modulation.

Usage

flux2_modulation(dim, mod_param_sets = 2L, bias = FALSE)

Arguments

  • dim: Integer. Model dimension.
  • mod_param_sets: Integer. Number of (shift, scale, gate) triples.
  • bias: Logical.