MCPcopy Create free account

hub / github.com/TencentARC/BrushNet / types & classes

Types & classes1,590 in github.com/TencentARC/BrushNet

↓ 155 callersClassOptionalDependencyNotAvailable
An error indicating that an optional dependency of Diffusers was not found in the environment.
src/diffusers/utils/import_utils.py:665
↓ 126 callersClassVaeImageProcessor
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 callersClassUNet2DConditionModel
src/diffusers/utils/dummy_pt_objects.py:245
↓ 79 callersClassFrozenDict
src/diffusers/configuration_utils.py:54
↓ 77 callersClassAutoencoderKL
src/diffusers/utils/dummy_pt_objects.py:20
↓ 74 callersClassDDIMScheduler
src/diffusers/utils/dummy_pt_objects.py:753
↓ 70 callersClassStableDiffusionPipelineOutput
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 callersClassResnetBlock2D
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 callersClass_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 callersClassStableDiffusionPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1085
↓ 43 callersClassImagePipelineOutput
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 callersClassPNDMScheduler
src/diffusers/utils/dummy_pt_objects.py:1023
↓ 38 callersClassTimestepEmbedding
src/diffusers/models/embeddings.py:191
↓ 36 callersClassAttention
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 callersClassLoRALinearLayer
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 callersClassAttnProcessor
r""" Default processor for performing attention-related computations.
src/diffusers/models/attention_processor.py:732
↓ 30 callersClassAttnAddedKVProcessor
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 callersClassConvResblock
scripts/convert_consistency_decoder.py:246
↓ 30 callersClassDDPMScheduler
src/diffusers/utils/dummy_pt_objects.py:783
↓ 28 callersClassCaptureLogger
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 callersClassStableDiffusionXLPipelineOutput
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 callersClassResConvBlock
src/diffusers/models/unets/unet_1d_blocks.py:375
↓ 24 callersClassTimesteps
src/diffusers/models/embeddings.py:240
↓ 24 callersClassUNet2DModel
src/diffusers/models/unet_2d.py:25
↓ 22 callersClassControlNetModel
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 callersClassTransformer2DModel
src/diffusers/models/transformer_2d.py:23
↓ 20 callersClassStableDiffusionXLImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1220
↓ 20 callersClassUpsample2D
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 callersClassVQModel
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 callersClassDownsample2D
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 callersClassMultiControlNetModel
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 callersClassDecoderOutput
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 callersClassEulerDiscreteScheduler
src/diffusers/utils/dummy_pt_objects.py:918
↓ 16 callersClassT2IAdapter
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 callersClassEMAModel
Exponential Moving Average of models weights
src/diffusers/training_utils.py:169
↓ 15 callersClassEvent
src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:116
↓ 15 callersClassSchedulerOutput
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 callersClassStableDiffusionXLInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1235
↓ 15 callersClassStableDiffusionXLPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1265
↓ 14 callersClassStableDiffusionXLWatermarker
src/diffusers/pipelines/stable_diffusion_xl/watermark.py:17
↓ 12 callersClassResnetBlockCondNorm2D
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 callersClassSelfAttention1d
src/diffusers/models/unets/unet_1d_blocks.py:317
↓ 11 callersClassConsistencyModelPipeline
src/diffusers/utils/dummy_pt_objects.py:483
↓ 11 callersClassDDPMPipeline
src/diffusers/utils/dummy_pt_objects.py:528
↓ 11 callersClassRMSNorm
src/diffusers/models/normalization.py:212
↓ 11 callersClassTransformerTemporalModel
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 callersClassUniDiffuserPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1415
↓ 10 callersClassAttnProcsLayers
src/diffusers/loaders/utils.py:20
↓ 10 callersClassDDPMWuerstchenScheduler
src/diffusers/utils/dummy_pt_objects.py:798
↓ 10 callersClassFeedForward
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 callersClassFlaxResnetBlock2D
Flax implementation of 2D Resnet Block. Args: in_channels (`int`): Input channels out_channels (`int`):
src/diffusers/models/vae_flax.py:125
↓ 10 callersClassResnetBlockFlat
src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1445
↓ 10 callersClassStableDiffusionImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:950
↓ 10 callersClassStableDiffusionInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:965
↓ 10 callersClassTransformer2DModel
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 callersClassAttnProcessor2_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 callersClassAudioLDM2Pipeline
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 callersClassAudioLDMPipeline
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 callersClassAudioPipelineOutput
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 callersClassCMStochasticIterativeScheduler
src/diffusers/utils/dummy_pt_objects.py:708
↓ 8 callersClassMusicLDMPipeline
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 callersClassResidualTemporalBlock1D
Residual 1D block with temporal convolutions. Parameters: inp_channels (`int`): Number of input channels. out_channels (`int
src/diffusers/models/resnet.py:456
↓ 8 callersClassSDCascadeLayerNorm
src/diffusers/models/unets/unet_stable_cascade.py:30
↓ 8 callersClassSpatioTemporalResBlock
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 callersClassUnCLIPScheduler
src/diffusers/utils/dummy_pt_objects.py:1113
↓ 8 callersClassWuerstchenLayerNorm
src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py:9
↓ 7 callersClassPriorTransformer
src/diffusers/utils/dummy_pt_objects.py:170
↓ 7 callersClassStableDiffusionPanoramaPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1055
↓ 7 callersClassStableDiffusionXLAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1160
↓ 6 callersClassAdaLayerNorm
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 callersClassBasicTransformerBlock
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 callersClassCustomOutput
tests/others/test_outputs.py:14
↓ 6 callersClassDPMSolverMultistepScheduler
src/diffusers/utils/dummy_pt_objects.py:843
↓ 6 callersClassDiagonalGaussianDistribution
src/diffusers/models/autoencoders/vae.py:769
↓ 6 callersClassDualTransformer2DModel
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 callersClassEventRange
src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:109
↓ 6 callersClassIFPipelineOutput
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 callersClassImagePipelineOutput
src/diffusers/utils/dummy_pt_objects.py:573
↓ 6 callersClassKandinskyPriorPipelineOutput
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 callersClassLinearMultiDim
src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1426
↓ 6 callersClassSampleObject
tests/others/test_config.py:32
↓ 6 callersClassStableCascadeUNet
src/diffusers/models/unets/unet_stable_cascade.py:137
↓ 6 callersClassStableDiffusionAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:800
↓ 6 callersClassStableDiffusionPipeline
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 callersClassTemporalConvLayer
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 callersClassUNet2DConditionModel
src/diffusers/models/unet_2d_condition.py:23
↓ 5 callersClassAdapterBlock
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 callersClassAnimateDiffPipelineOutput
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 callersClassDownsample1D
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 callersClassEncoder
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 callersClassLoRACompatibleConv
A convolutional layer that can be used with LoRA.
src/diffusers/models/lora.py:293
↓ 5 callersClassMotionAdapter
src/diffusers/utils/dummy_pt_objects.py:140
↓ 5 callersClassMultiAdapter
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 callersClassPaellaVQModel
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 callersClassSpatialNorm
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 callersClassStableDiffusionInstructPix2PixPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:995
↓ 5 callersClassStableDiffusionUpscalePipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1145
↓ 5 callersClassUNetMidBlock2D
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 callersClassAdaGroupNorm
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 callersClassAttnAddedKVProcessor2_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