Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/TencentARC/BrushNet
/ types & classes
Types & classes
1,590 in github.com/TencentARC/BrushNet
⨍
Functions
9,976
◇
Types & classes
1,590
↳
Endpoints
3
↓ 155 callers
Class
OptionalDependencyNotAvailable
An error indicating that an optional dependency of Diffusers was not found in the environment.
src/diffusers/utils/import_utils.py:665
↓ 126 callers
Class
VaeImageProcessor
Image processor for VAE. Args: do_resize (`bool`, *optional*, defaults to `True`): Whether to downscale the image's (hei
src/diffusers/image_processor.py:41
↓ 90 callers
Class
UNet2DConditionModel
src/diffusers/utils/dummy_pt_objects.py:245
↓ 79 callers
Class
FrozenDict
src/diffusers/configuration_utils.py:54
↓ 77 callers
Class
AutoencoderKL
src/diffusers/utils/dummy_pt_objects.py:20
↓ 74 callers
Class
DDIMScheduler
src/diffusers/utils/dummy_pt_objects.py:753
↓ 70 callers
Class
StableDiffusionPipelineOutput
Output class for Stable Diffusion pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL
src/diffusers/pipelines/stable_diffusion/pipeline_output.py:11
↓ 59 callers
Class
ResnetBlock2D
r""" A Resnet block. Parameters: in_channels (`int`): The number of channels in the input. out_channels (`int`, *optional*, d
src/diffusers/models/resnet.py:195
↓ 59 callers
Class
_LazyModule
Module class that surfaces all objects but only performs associated imports when the objects are requested.
src/diffusers/utils/import_utils.py:669
↓ 48 callers
Class
StableDiffusionPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1085
↓ 43 callers
Class
ImagePipelineOutput
Output class for image pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL images of
src/diffusers/pipelines/pipeline_utils.py:97
↓ 38 callers
Class
PNDMScheduler
src/diffusers/utils/dummy_pt_objects.py:1023
↓ 38 callers
Class
TimestepEmbedding
src/diffusers/models/embeddings.py:191
↓ 36 callers
Class
Attention
r""" A cross attention layer. Parameters: query_dim (`int`): The number of channels in the query. cross_attention
src/diffusers/models/attention_processor.py:40
↓ 36 callers
Class
LoRALinearLayer
r""" A linear layer that is used with LoRA. Parameters: in_features (`int`): Number of input features. out_featur
src/diffusers/models/lora.py:175
↓ 32 callers
Class
AttnProcessor
r""" Default processor for performing attention-related computations.
src/diffusers/models/attention_processor.py:732
↓ 30 callers
Class
AttnAddedKVProcessor
r""" Processor for performing attention-related computations with extra learnable key and value matrices for the text encoder.
src/diffusers/models/attention_processor.py:905
↓ 30 callers
Class
ConvResblock
scripts/convert_consistency_decoder.py:246
↓ 30 callers
Class
DDPMScheduler
src/diffusers/utils/dummy_pt_objects.py:783
↓ 28 callers
Class
CaptureLogger
Args: Context manager to capture `logging` streams logger: 'logging` logger object Returns: The captured output is availa
src/diffusers/utils/testing_utils.py:765
↓ 27 callers
Class
StableDiffusionXLPipelineOutput
Output class for Stable Diffusion pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL
src/diffusers/pipelines/stable_diffusion_xl/pipeline_output.py:11
↓ 24 callers
Class
ResConvBlock
src/diffusers/models/unets/unet_1d_blocks.py:375
↓ 24 callers
Class
Timesteps
src/diffusers/models/embeddings.py:240
↓ 24 callers
Class
UNet2DModel
src/diffusers/models/unet_2d.py:25
↓ 22 callers
Class
ControlNetModel
A ControlNet model. Args: in_channels (`int`, defaults to 4): The number of channels in the input sample. flip_s
src/diffusers/models/controlnet.py:111
↓ 21 callers
Class
Transformer2DModel
src/diffusers/models/transformer_2d.py:23
↓ 20 callers
Class
StableDiffusionXLImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1220
↓ 20 callers
Class
Upsample2D
A 2D upsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and output
src/diffusers/models/upsampling.py:76
↓ 20 callers
Class
VQModel
r""" A VQ-VAE model for decoding latent representations. This model inherits from [`ModelMixin`]. Check the superclass documentation for it's
src/diffusers/models/vq_model.py:40
↓ 19 callers
Class
Downsample2D
A 2D downsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and outp
src/diffusers/models/downsampling.py:70
↓ 19 callers
Class
MultiControlNetModel
r""" Multiple `ControlNetModel` wrapper class for Multi-ControlNet This module is a wrapper for multiple instances of the `ControlNetModel`.
src/diffusers/pipelines/controlnet/multicontrolnet.py:15
↓ 17 callers
Class
DecoderOutput
r""" Output of decoding method. Args: sample (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`):
src/diffusers/models/autoencoders/vae.py:34
↓ 17 callers
Class
EulerDiscreteScheduler
src/diffusers/utils/dummy_pt_objects.py:918
↓ 16 callers
Class
T2IAdapter
r""" A simple ResNet-like model that accepts images containing control signals such as keyposes and depth. The model generates multiple featur
src/diffusers/models/adapter.py:217
↓ 15 callers
Class
EMAModel
Exponential Moving Average of models weights
src/diffusers/training_utils.py:169
↓ 15 callers
Class
Event
src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:116
↓ 15 callers
Class
SchedulerOutput
Base class for the output of a scheduler's `step` function. Args: prev_sample (`torch.FloatTensor` of shape `(batch_size, num_channe
src/diffusers/schedulers/scheduling_utils.py:52
↓ 15 callers
Class
StableDiffusionXLInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1235
↓ 15 callers
Class
StableDiffusionXLPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1265
↓ 14 callers
Class
StableDiffusionXLWatermarker
src/diffusers/pipelines/stable_diffusion_xl/watermark.py:17
↓ 12 callers
Class
ResnetBlockCondNorm2D
r""" A Resnet block that use normalization layer that incorporate conditioning information. Parameters: in_channels (`int`): The numb
src/diffusers/models/resnet.py:45
↓ 12 callers
Class
SelfAttention1d
src/diffusers/models/unets/unet_1d_blocks.py:317
↓ 11 callers
Class
ConsistencyModelPipeline
src/diffusers/utils/dummy_pt_objects.py:483
↓ 11 callers
Class
DDPMPipeline
src/diffusers/utils/dummy_pt_objects.py:528
↓ 11 callers
Class
RMSNorm
src/diffusers/models/normalization.py:212
↓ 11 callers
Class
TransformerTemporalModel
A Transformer model for video-like data. Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads to
src/diffusers/models/transformers/transformer_temporal.py:41
↓ 11 callers
Class
UniDiffuserPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1415
↓ 10 callers
Class
AttnProcsLayers
src/diffusers/loaders/utils.py:20
↓ 10 callers
Class
DDPMWuerstchenScheduler
src/diffusers/utils/dummy_pt_objects.py:798
↓ 10 callers
Class
FeedForward
r""" A feed-forward layer. Parameters: dim (`int`): The number of channels in the input. dim_out (`int`, *optional*): The num
src/diffusers/models/attention.py:617
↓ 10 callers
Class
FlaxResnetBlock2D
Flax implementation of 2D Resnet Block. Args: in_channels (`int`): Input channels out_channels (`int`):
src/diffusers/models/vae_flax.py:125
↓ 10 callers
Class
ResnetBlockFlat
src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1445
↓ 10 callers
Class
StableDiffusionImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:950
↓ 10 callers
Class
StableDiffusionInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:965
↓ 10 callers
Class
Transformer2DModel
A 2D Transformer model for image-like data. Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads
src/diffusers/models/transformers/transformer_2d.py:44
↓ 8 callers
Class
AttnProcessor2_0
r""" Processor for implementing scaled dot-product attention (enabled by default if you're using PyTorch 2.0).
src/diffusers/models/attention_processor.py:1204
↓ 8 callers
Class
AudioLDM2Pipeline
r""" Pipeline for text-to-audio generation using AudioLDM2. This model inherits from [`DiffusionPipeline`]. Check the superclass documentatio
src/diffusers/pipelines/audioldm2/pipeline_audioldm2.py:103
↓ 8 callers
Class
AudioLDMPipeline
r""" Pipeline for text-to-audio generation using AudioLDM. This model inherits from [`DiffusionPipeline`]. Check the superclass documentation
src/diffusers/pipelines/audioldm/pipeline_audioldm.py:52
↓ 8 callers
Class
AudioPipelineOutput
Output class for audio pipelines. Args: audios (`np.ndarray`) List of denoised audio samples of a NumPy array of shape `
src/diffusers/pipelines/pipeline_utils.py:111
↓ 8 callers
Class
CMStochasticIterativeScheduler
src/diffusers/utils/dummy_pt_objects.py:708
↓ 8 callers
Class
MusicLDMPipeline
r""" Pipeline for text-to-audio generation using MusicLDM. This model inherits from [`DiffusionPipeline`]. Check the superclass documentation
src/diffusers/pipelines/musicldm/pipeline_musicldm.py:67
↓ 8 callers
Class
ResidualTemporalBlock1D
Residual 1D block with temporal convolutions. Parameters: inp_channels (`int`): Number of input channels. out_channels (`int
src/diffusers/models/resnet.py:456
↓ 8 callers
Class
SDCascadeLayerNorm
src/diffusers/models/unets/unet_stable_cascade.py:30
↓ 8 callers
Class
SpatioTemporalResBlock
r""" A SpatioTemporal Resnet block. Parameters: in_channels (`int`): The number of channels in the input. out_channels (`int`
src/diffusers/models/resnet.py:667
↓ 8 callers
Class
UnCLIPScheduler
src/diffusers/utils/dummy_pt_objects.py:1113
↓ 8 callers
Class
WuerstchenLayerNorm
src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py:9
↓ 7 callers
Class
PriorTransformer
src/diffusers/utils/dummy_pt_objects.py:170
↓ 7 callers
Class
StableDiffusionPanoramaPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1055
↓ 7 callers
Class
StableDiffusionXLAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1160
↓ 6 callers
Class
AdaLayerNorm
r""" Norm layer modified to incorporate timestep embeddings. Parameters: embedding_dim (`int`): The size of each embedding vector.
src/diffusers/models/normalization.py:28
↓ 6 callers
Class
BasicTransformerBlock
r""" A basic Transformer block. Parameters: dim (`int`): The number of channels in the input and output. num_attention_heads
src/diffusers/models/attention.py:97
↓ 6 callers
Class
CustomOutput
tests/others/test_outputs.py:14
↓ 6 callers
Class
DPMSolverMultistepScheduler
src/diffusers/utils/dummy_pt_objects.py:843
↓ 6 callers
Class
DiagonalGaussianDistribution
src/diffusers/models/autoencoders/vae.py:769
↓ 6 callers
Class
DualTransformer2DModel
Dual transformer wrapper that combines two `Transformer2DModel`s for mixed inference. Parameters: num_attention_heads (`int`, *optio
src/diffusers/models/transformers/dual_transformer_2d.py:21
↓ 6 callers
Class
EventRange
src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:109
↓ 6 callers
Class
IFPipelineOutput
Args: Output class for Stable Diffusion pipelines. images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL
src/diffusers/pipelines/deepfloyd_if/pipeline_output.py:11
↓ 6 callers
Class
ImagePipelineOutput
src/diffusers/utils/dummy_pt_objects.py:573
↓ 6 callers
Class
KandinskyPriorPipelineOutput
Output class for KandinskyPriorPipeline. Args: image_embeds (`torch.FloatTensor`) clip image embeddings for text prompt
src/diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py:113
↓ 6 callers
Class
LinearMultiDim
src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1426
↓ 6 callers
Class
SampleObject
tests/others/test_config.py:32
↓ 6 callers
Class
StableCascadeUNet
src/diffusers/models/unets/unet_stable_cascade.py:137
↓ 6 callers
Class
StableDiffusionAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:800
↓ 6 callers
Class
StableDiffusionPipeline
r""" Pipeline for text-to-image generation using Stable Diffusion. This model inherits from [`DiffusionPipeline`]. Check the superclass docum
examples/community/sd_text2img_k_diffusion.py:44
↓ 6 callers
Class
TemporalConvLayer
Temporal convolutional layer that can be used for video (sequence of images) input Code mostly copied from: https://github.com/modelscope/mod
src/diffusers/models/resnet.py:503
↓ 6 callers
Class
UNet2DConditionModel
src/diffusers/models/unet_2d_condition.py:23
↓ 5 callers
Class
AdapterBlock
r""" An AdapterBlock is a helper model that contains multiple ResNet-like blocks. It is used in the `FullAdapter` and `FullAdapterXL` models.
src/diffusers/models/adapter.py:391
↓ 5 callers
Class
AnimateDiffPipelineOutput
r""" Output class for AnimateDiff pipelines. Args: frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
src/diffusers/pipelines/animatediff/pipeline_output.py:12
↓ 5 callers
Class
Downsample1D
A 1D downsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and outp
src/diffusers/models/downsampling.py:27
↓ 5 callers
Class
Encoder
r""" The `Encoder` layer of a variational autoencoder that encodes its input into a latent representation. Args: in_channels (`int`,
src/diffusers/models/autoencoders/vae.py:46
↓ 5 callers
Class
LoRACompatibleConv
A convolutional layer that can be used with LoRA.
src/diffusers/models/lora.py:293
↓ 5 callers
Class
MotionAdapter
src/diffusers/utils/dummy_pt_objects.py:140
↓ 5 callers
Class
MultiAdapter
r""" MultiAdapter is a wrapper model that contains multiple adapter models and merges their outputs according to user-assigned weighting.
src/diffusers/models/adapter.py:28
↓ 5 callers
Class
PaellaVQModel
r"""VQ-VAE model from Paella model. This model inherits from [`ModelMixin`]. Check the superclass documentation for the generic methods the libra
src/diffusers/pipelines/wuerstchen/modeling_paella_vq_model.py:58
↓ 5 callers
Class
SpatialNorm
Spatially conditioned normalization as defined in https://arxiv.org/abs/2209.09002. Args: f_channels (`int`): The number
src/diffusers/models/attention_processor.py:1795
↓ 5 callers
Class
StableDiffusionInstructPix2PixPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:995
↓ 5 callers
Class
StableDiffusionUpscalePipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1145
↓ 5 callers
Class
UNetMidBlock2D
A 2D UNet mid-block [`UNetMidBlock2D`] with multiple residual blocks and optional attention blocks. Args: in_channels (`int`): The n
src/diffusers/models/unets/unet_2d_blocks.py:601
↓ 4 callers
Class
AdaGroupNorm
r""" GroupNorm layer modified to incorporate timestep embeddings. Parameters: embedding_dim (`int`): The size of each embedding vecto
src/diffusers/models/normalization.py:115
↓ 4 callers
Class
AttnAddedKVProcessor2_0
r""" Processor for performing scaled dot-product attention (enabled by default if you're using PyTorch 2.0), with extra learnable key and valu
src/diffusers/models/attention_processor.py:969
next →
1–100 of 1,590, ranked by callers