Python module

max.pipelines.architectures.flux2_modulev3

FLUX.2 diffusion architecture for image generation.

Flux2TransformerModel

class max.pipelines.architectures.flux2_modulev3.Flux2TransformerModel(config, encoding, devices, weights, *, cache_config=None)

source

Bases: ComponentModel

Parameters:

• config (dict[str, Any])
• encoding (SupportedEncoding)
• devices (list[Device])
• weights (Weights)
• cache_config (DenoisingCacheConfig | None)

load_model()

load_model()

source

Load and return a runtime model instance.

Return type:

None

model

model: Callable[[...], Any]

source

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit