Types & classes — github.com/black-forest-labs/flux

CodeHub MCPcopy Create free account

hub / github.com/black-forest-labs/flux / types & classes

Types & classes54 in github.com/black-forest-labs/flux

⨍ Functions 217◇ Types & classes 54

↓ 10 callersClassAutoEncoderParams

src/flux/modules/autoencoder.py:9

↓ 10 callersClassFluxParams

src/flux/model.py:18

↓ 10 callersClassModelSpec

src/flux/util.py:288

↓ 6 callersClassResnetBlock

src/flux/modules/autoencoder.py:55

↓ 3 callersClassMLPEmbedder

src/flux/modules/layers.py:52

↓ 3 callersClassModulation

src/flux/modules/layers.py:113

↓ 3 callersClassSamplingOptions

src/flux/cli.py:26

↓ 3 callersClassTRTManager

src/flux/trt/trt_manager.py:46

↓ 2 callersClassAttnBlock

src/flux/modules/autoencoder.py:25

↓ 2 callersClassHFEmbedder

src/flux/modules/conditioner.py:5

↓ 2 callersClassModulationOut

src/flux/modules/layers.py:107

↓ 2 callersClassQKNorm

src/flux/modules/layers.py:75

↓ 2 callersClassRMSNorm

src/flux/modules/layers.py:63

↓ 2 callersClassSelfAttention

src/flux/modules/layers.py:87

↓ 1 callersClassAutoEncoder

src/flux/modules/autoencoder.py:282

↓ 1 callersClassCannyImageEncoder

src/flux/modules/image_embedders.py:36

↓ 1 callersClassDecoder

src/flux/modules/autoencoder.py:183

↓ 1 callersClassDepthImageEncoder

src/flux/modules/image_embedders.py:13

↓ 1 callersClassDiagonalGaussian

src/flux/modules/autoencoder.py:267

↓ 1 callersClassDoubleStreamBlock

src/flux/modules/layers.py:129

↓ 1 callersClassDownsample

src/flux/modules/autoencoder.py:85

↓ 1 callersClassEmbedND

src/flux/modules/layers.py:11

↓ 1 callersClassEncoder

src/flux/modules/autoencoder.py:109

↓ 1 callersClassFlux

Transformer model for flow matching on sequences.

src/flux/model.py:34

↓ 1 callersClassFluxGenerator

↓ 1 callersClassFluxLoraWrapper

src/flux/model.py:122

↓ 1 callersClassLastLayer

src/flux/modules/layers.py:242

↓ 1 callersClassLinearLora

src/flux/modules/lora.py:34

↓ 1 callersClassModuleName

src/flux/trt/trt_config/base_trt_config.py:30

↓ 1 callersClassPixtralContentFilter

src/flux/content_filters.py:59

↓ 1 callersClassReduxImageEncoder

src/flux/modules/image_embedders.py:64

↓ 1 callersClassSamplingOptions

src/flux/cli_fill.py:17

↓ 1 callersClassSamplingOptions

src/flux/cli_kontext.py:24

↓ 1 callersClassSamplingOptions

src/flux/cli_control.py:17

↓ 1 callersClassSamplingOptions

src/flux/cli_redux.py:24

↓ 1 callersClassSharedMemory

src/flux/trt/engine/base_engine.py:32

↓ 1 callersClassSingleStreamBlock

A DiT block with parallel linear layers as described in https://arxiv.org/abs/2302.05442 and adapted modulation interface.

src/flux/modules/layers.py:194

↓ 1 callersClassUpsample

src/flux/modules/autoencoder.py:98

↓ 1 callersClassVAEEngine

src/flux/trt/engine/vae_engine.py:54

↓ 1 callersClassWatermarkEmbedder

src/flux/util.py:733

ClassBaseEngine

src/flux/trt/engine/base_engine.py:99

ClassCLIPEngine

src/flux/trt/engine/clip_engine.py:24

ClassClipConfig

src/flux/trt/trt_config/clip_trt_config.py:25

src/flux/trt/engine/base_engine.py:130

src/flux/trt/trt_config/t5_trt_config.py:29

src/flux/trt/engine/t5_engine.py:24

ClassTRTBaseConfig

src/flux/trt/trt_config/base_trt_config.py:42

ClassTransformerConfig

src/flux/trt/trt_config/transformer_trt_config.py:31

ClassTransformerEngine

src/flux/trt/engine/transformer_engine.py:23

ClassVAEBaseConfig

src/flux/trt/trt_config/vae_trt_config.py:25

ClassVAEDecoder

src/flux/trt/engine/vae_engine.py:23

ClassVAEDecoderConfig

src/flux/trt/trt_config/vae_trt_config.py:71

ClassVAEEncoder

src/flux/trt/engine/vae_engine.py:39

ClassVAEEncoderConfig

src/flux/trt/trt_config/vae_trt_config.py:175