Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/TencentARC/FluxKits
/ types & classes
Types & classes
86 in github.com/TencentARC/FluxKits
⨍
Functions
314
◇
Types & classes
86
↓ 8 callers
Class
LoRALinearLayer
flux-mini/src/flux/modules/layers.py:85
↓ 8 callers
Class
LoRALinearLayer
flux-npu/src/flux/modules/layers.py:85
↓ 6 callers
Class
MLPEmbedder
flux-mini/src/flux/modules/layers.py:51
↓ 6 callers
Class
MLPEmbedder
flux-npu/src/flux/modules/layers.py:51
↓ 6 callers
Class
ResnetBlock
flux-mini/src/flux/modules/autoencoder.py:55
↓ 6 callers
Class
ResnetBlock
flux-npu/src/flux/modules/autoencoder.py:55
↓ 5 callers
Class
AutoEncoderParams
flux-mini/src/flux/modules/autoencoder.py:9
↓ 4 callers
Class
ApiException
flux-mini/src/flux/api.py:12
↓ 4 callers
Class
ApiException
flux-npu/src/flux/api.py:12
↓ 4 callers
Class
AutoEncoderParams
flux-npu/src/flux/modules/autoencoder.py:9
↓ 4 callers
Class
FluxParams
flux-mini/src/flux/model.py:20
↓ 4 callers
Class
FluxParams
flux-npu/src/flux/model.py:20
↓ 4 callers
Class
ModelSpec
flux-mini/src/flux/util.py:153
↓ 4 callers
Class
ModelSpec
flux-npu/src/flux/util.py:153
↓ 3 callers
Class
Flux
Transformer model for flow matching on sequences.
flux-mini/src/flux/model.py:35
↓ 3 callers
Class
Flux
Transformer model for flow matching on sequences.
flux-npu/src/flux/model.py:35
↓ 3 callers
Class
Modulation
flux-mini/src/flux/modules/layers.py:161
↓ 3 callers
Class
Modulation
flux-npu/src/flux/modules/layers.py:161
↓ 2 callers
Class
AttnBlock
flux-mini/src/flux/modules/autoencoder.py:25
↓ 2 callers
Class
AttnBlock
flux-npu/src/flux/modules/autoencoder.py:25
↓ 2 callers
Class
AutoEncoder
flux-mini/src/flux/modules/autoencoder.py:277
↓ 2 callers
Class
DoubleStreamBlock
flux-mini/src/flux/modules/layers.py:339
↓ 2 callers
Class
DoubleStreamBlock
flux-npu/src/flux/modules/layers.py:339
↓ 2 callers
Class
DoubleStreamBlockLoraProcessor
flux-mini/src/flux/modules/layers.py:176
↓ 2 callers
Class
DoubleStreamBlockLoraProcessor
flux-npu/src/flux/modules/layers.py:176
↓ 2 callers
Class
EmbedND
flux-mini/src/flux/modules/layers.py:12
↓ 2 callers
Class
EmbedND
flux-npu/src/flux/modules/layers.py:12
↓ 2 callers
Class
HFEmbedder
flux-mini/src/flux/modules/conditioner.py:6
↓ 2 callers
Class
HFEmbedder
flux-npu/src/flux/modules/conditioner.py:7
↓ 2 callers
Class
ModulationOut
flux-mini/src/flux/modules/layers.py:155
↓ 2 callers
Class
ModulationOut
flux-npu/src/flux/modules/layers.py:155
↓ 2 callers
Class
QKNorm
flux-mini/src/flux/modules/layers.py:74
↓ 2 callers
Class
QKNorm
flux-npu/src/flux/modules/layers.py:74
↓ 2 callers
Class
RMSNorm
flux-mini/src/flux/modules/layers.py:62
↓ 2 callers
Class
RMSNorm
flux-npu/src/flux/modules/layers.py:62
↓ 2 callers
Class
SelfAttention
flux-mini/src/flux/modules/layers.py:141
↓ 2 callers
Class
SelfAttention
flux-npu/src/flux/modules/layers.py:141
↓ 2 callers
Class
SingleStreamBlockProcessor
flux-mini/src/flux/modules/layers.py:488
↓ 2 callers
Class
SingleStreamBlockProcessor
flux-npu/src/flux/modules/layers.py:488
↓ 1 callers
Class
Annotator
flux-mini/src/flux/util.py:114
↓ 1 callers
Class
Annotator
flux-npu/src/flux/util.py:114
↓ 1 callers
Class
AutoEncoder
flux-npu/src/flux/modules/autoencoder.py:277
↓ 1 callers
Class
ControlNetFlux
Transformer model for flow matching on sequences.
flux-mini/src/flux/controlnet.py:33
↓ 1 callers
Class
ControlNetFlux
Transformer model for flow matching on sequences.
flux-npu/src/flux/controlnet.py:32
↓ 1 callers
Class
CustomImageDataset
flux-mini/image_datasets/dataset.py:54
↓ 1 callers
Class
CustomImageDataset
flux-npu/image_datasets/dataset.py:54
↓ 1 callers
Class
Decoder
flux-mini/src/flux/modules/autoencoder.py:183
↓ 1 callers
Class
Decoder
flux-npu/src/flux/modules/autoencoder.py:183
↓ 1 callers
Class
DiTDistiller
flux-mini/train_flux_distill.py:136
↓ 1 callers
Class
DiTDistiller
flux-npu/train_flux_distill.py:138
↓ 1 callers
Class
DiagonalGaussian
flux-mini/src/flux/modules/autoencoder.py:262
↓ 1 callers
Class
DiagonalGaussian
flux-npu/src/flux/modules/autoencoder.py:262
↓ 1 callers
Class
DoubleStreamBlockProcessor
flux-mini/src/flux/modules/layers.py:300
↓ 1 callers
Class
DoubleStreamBlockProcessor
flux-npu/src/flux/modules/layers.py:300
↓ 1 callers
Class
Downsample
flux-mini/src/flux/modules/autoencoder.py:85
↓ 1 callers
Class
Downsample
flux-npu/src/flux/modules/autoencoder.py:85
↓ 1 callers
Class
Encoder
flux-mini/src/flux/modules/autoencoder.py:109
↓ 1 callers
Class
Encoder
flux-npu/src/flux/modules/autoencoder.py:109
↓ 1 callers
Class
IPDoubleStreamBlockProcessor
Attention processor for handling IP-adapter with double stream block.
flux-mini/src/flux/modules/layers.py:222
↓ 1 callers
Class
IPDoubleStreamBlockProcessor
Attention processor for handling IP-adapter with double stream block.
flux-npu/src/flux/modules/layers.py:222
↓ 1 callers
Class
ImageProjModel
Projection Model https://github.com/tencent-ailab/IP-Adapter/blob/main/ip_adapter/ip_adapter.py#L28
flux-mini/src/flux/modules/layers.py:579
↓ 1 callers
Class
ImageProjModel
Projection Model https://github.com/tencent-ailab/IP-Adapter/blob/main/ip_adapter/ip_adapter.py#L28
flux-npu/src/flux/modules/layers.py:579
↓ 1 callers
Class
LastLayer
flux-mini/src/flux/modules/layers.py:565
↓ 1 callers
Class
LastLayer
flux-npu/src/flux/modules/layers.py:565
↓ 1 callers
Class
SamplingOptions
flux-mini/src/flux/cli.py:19
↓ 1 callers
Class
SamplingOptions
flux-npu/src/flux/cli.py:21
↓ 1 callers
Class
SingleStreamBlock
A DiT block with parallel linear layers as described in https://arxiv.org/abs/2302.05442 and adapted modulation interface.
flux-mini/src/flux/modules/layers.py:508
↓ 1 callers
Class
SingleStreamBlock
A DiT block with parallel linear layers as described in https://arxiv.org/abs/2302.05442 and adapted modulation interface.
flux-npu/src/flux/modules/layers.py:508
↓ 1 callers
Class
SingleStreamBlockLoraProcessor
flux-mini/src/flux/modules/layers.py:460
↓ 1 callers
Class
SingleStreamBlockLoraProcessor
flux-npu/src/flux/modules/layers.py:460
↓ 1 callers
Class
Upsample
flux-mini/src/flux/modules/autoencoder.py:98
↓ 1 callers
Class
Upsample
flux-npu/src/flux/modules/autoencoder.py:98
↓ 1 callers
Class
WatermarkEmbedder
flux-mini/src/flux/util.py:432
↓ 1 callers
Class
WatermarkEmbedder
flux-npu/src/flux/util.py:432
Class
FLuxSelfAttnProcessor
flux-mini/src/flux/modules/layers.py:111
Class
FLuxSelfAttnProcessor
flux-npu/src/flux/modules/layers.py:111
Class
FluxParams
flux-mini/src/flux/controlnet.py:13
Class
FluxParams
flux-npu/src/flux/controlnet.py:12
Class
IPSingleStreamBlockProcessor
Attention processor for handling IP-adapter with single stream block.
flux-mini/src/flux/modules/layers.py:393
Class
IPSingleStreamBlockProcessor
Attention processor for handling IP-adapter with single stream block.
flux-npu/src/flux/modules/layers.py:393
Class
ImageRequest
flux-mini/src/flux/api.py:31
Class
ImageRequest
flux-npu/src/flux/api.py:31
Class
LoraFluxAttnProcessor
flux-mini/src/flux/modules/layers.py:122
Class
LoraFluxAttnProcessor
flux-npu/src/flux/modules/layers.py:122
Class
XFluxPipeline
flux-mini/src/flux/xflux_pipeline.py:32
Class
XFluxPipeline
flux-npu/src/flux/xflux_pipeline.py:31