MCPcopy Create free account

hub / github.com/LetterLiGo/SafeGen_CCS2024 / types & classes

Types & classes1,211 in github.com/LetterLiGo/SafeGen_CCS2024

↓ 123 callersClassOptionalDependencyNotAvailable
An error indicating that an optional dependency of Diffusers was not found in the environment.
src/diffusers/utils/import_utils.py:657
↓ 97 callersClassUNet2DConditionModel
src/diffusers/utils/dummy_pt_objects.py:230
↓ 86 callersClassAutoencoderKL
r""" A VAE model with KL loss for encoding images into latents and decoding latent representations into images. This model inherits from [`Mo
src/diffusers/models/autoencoder_kl.py:34
↓ 77 callersClassDDIMScheduler
src/diffusers/utils/dummy_pt_objects.py:693
↓ 75 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:46
↓ 61 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:584
↓ 59 callersClassStableDiffusionPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:950
↓ 47 callersClassPNDMScheduler
src/diffusers/utils/dummy_pt_objects.py:933
↓ 45 callersClass_LazyModule
Module class that surfaces all objects but only performs associated imports when the objects are requested.
src/diffusers/utils/import_utils.py:661
↓ 41 callersClassFrozenDict
src/diffusers/configuration_utils.py:50
↓ 37 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:111
↓ 32 callersClassTimestepEmbedding
src/diffusers/models/embeddings.py:190
↓ 32 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/transformer_2d.py:45
↓ 31 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:38
↓ 30 callersClassConvResblock
scripts/convert_consistency_decoder.py:246
↓ 29 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
↓ 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:708
↓ 28 callersClassUNet2DModel
r""" A 2D UNet model that takes a noisy sample and a timestep and returns a sample shaped output. This model inherits from [`ModelMixin`]. Ch
src/diffusers/models/unet_2d.py:40
↓ 27 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:869
↓ 27 callersClassAttnProcessor
r""" Default processor for performing attention-related computations.
src/diffusers/models/attention_processor.py:696
↓ 27 callersClassDDPMScheduler
src/diffusers/utils/dummy_pt_objects.py:723
↓ 27 callersClassStableDiffusionXLPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1130
↓ 24 callersClassResConvBlock
src/diffusers/models/unet_1d_blocks.py:375
↓ 21 callersClassTimesteps
src/diffusers/models/embeddings.py:238
↓ 20 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:105
↓ 20 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:172
↓ 19 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
↓ 18 callersClassEulerDiscreteScheduler
src/diffusers/utils/dummy_pt_objects.py:828
↓ 18 callersClassStableDiffusionXLImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1085
↓ 18 callersClassUpsample2D
A 2D upsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and output
src/diffusers/models/resnet.py:123
↓ 17 callersClassDecoderOutput
r""" Output of decoding method. Args: sample (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`):
src/diffusers/models/vae.py:34
↓ 17 callersClassDownsample2D
A 2D downsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and outp
src/diffusers/models/resnet.py:216
↓ 15 callersClassEvent
src/diffusers/pipelines/spectrogram_diffusion/midi_utils.py:116
↓ 15 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
↓ 14 callersClassStableDiffusionXLInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1100
↓ 14 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
↓ 13 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:50
↓ 12 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
↓ 12 callersClassSelfAttention1d
src/diffusers/models/unet_1d_blocks.py:317
↓ 11 callersClassUniDiffuserPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1280
↓ 10 callersClassDualTransformer2DModel
Dual transformer wrapper that combines two `Transformer2DModel`s for mixed inference. Parameters: num_attention_heads (`int`, *optio
src/diffusers/models/dual_transformer_2d.py:21
↓ 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/versatile_diffusion/modeling_text_unet.py:1400
↓ 10 callersClassStableDiffusionInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:830
↓ 10 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/transformer_temporal.py:41
↓ 9 callersClassConsistencyModelPipeline
src/diffusers/utils/dummy_pt_objects.py:453
↓ 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:125
↓ 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:850
↓ 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:1206
↓ 8 callersClassStableDiffusionImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:815
↓ 8 callersClassStableDiffusionPix2PixZeroPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:980
↓ 8 callersClassUnCLIPScheduler
src/diffusers/utils/dummy_pt_objects.py:993
↓ 8 callersClassWuerstchenLayerNorm
src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py:24
↓ 7 callersClassDDPMPipeline
src/diffusers/utils/dummy_pt_objects.py:498
↓ 7 callersClassPriorTransformer
src/diffusers/utils/dummy_pt_objects.py:155
↓ 7 callersClassStableDiffusionPanoramaPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:920
↓ 7 callersClassStableDiffusionXLAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1025
↓ 7 callersClassStableDiffusionXLWatermarker
src/diffusers/pipelines/stable_diffusion_xl/watermark.py:17
↓ 7 callersClassUNetMidBlock2D
A 2D UNet mid-block [`UNetMidBlock2D`] with multiple residual blocks and optional attention blocks. Args: in_channels (`int`): The n
src/diffusers/models/unet_2d_blocks.py:505
↓ 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:26
↓ 6 callersClassAttnProcsLayers
src/diffusers/loaders/utils.py:20
↓ 6 callersClassCMStochasticIterativeScheduler
src/diffusers/utils/dummy_pt_objects.py:648
↓ 6 callersClassCustomOutput
tests/others/test_outputs.py:14
↓ 6 callersClassDPMSolverMultistepScheduler
src/diffusers/utils/dummy_pt_objects.py:783
↓ 6 callersClassDiagonalGaussianDistribution
src/diffusers/models/vae.py:765
↓ 6 callersClassEventRange
src/diffusers/pipelines/spectrogram_diffusion/midi_utils.py:109
↓ 6 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:493
↓ 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 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/versatile_diffusion/modeling_text_unet.py:1381
↓ 6 callersClassSampleObject
tests/others/test_config.py:32
↓ 6 callersClassStableDiffusionAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:665
↓ 6 callersClassT5LayerNorm
r""" T5 style layer normalization module. Args: hidden_size (`int`): Size of the input hidden states. eps (`float
src/diffusers/models/t5_film_transformer.py:374
↓ 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:1042
↓ 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 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:1168
↓ 5 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:96
↓ 5 callersClassDownsample1D
A 1D downsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and outp
src/diffusers/models/resnet.py:80
↓ 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/vae.py:46
↓ 5 callersClassEulerAncestralDiscreteScheduler
src/diffusers/utils/dummy_pt_objects.py:813
↓ 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 callersClassResnetDownsampleBlock2D
src/diffusers/models/unet_2d_blocks.py:1610
↓ 5 callersClassResnetUpsampleBlock2D
src/diffusers/models/unet_2d_blocks.py:2861
↓ 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:1665
↓ 5 callersClassStableDiffusionInpaintPipelineLegacy
src/diffusers/utils/dummy_torch_and_transformers_objects.py:845
↓ 5 callersClassStableDiffusionInstructPix2PixPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:860
↓ 5 callersClassStableDiffusionUpscalePipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1010
↓ 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:113
↓ 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:933
↓ 4 callersClassAutoencoderKLOutput
Output of AutoencoderKL encoding method. Args: latent_dist (`DiagonalGaussianDistribution`): Encoded outputs of `Encoder
src/diffusers/models/modeling_outputs.py:7
↓ 4 callersClassCustomDiffusionAttnProcessor
r""" Processor for implementing attention for the Custom Diffusion method. Args: train_kv (`bool`, defaults to `True`): W
src/diffusers/models/attention_processor.py:765
↓ 4 callersClassDDPMWuerstchenScheduler
src/diffusers/utils/dummy_pt_objects.py:738
↓ 4 callersClassEMAModel
Exponential Moving Average of models weights
src/diffusers/training_utils.py:76
↓ 4 callersClassFlaxStableDiffusionPipelineOutput
Output class for Flax-based Stable Diffusion pipelines. Args: images (`np.ndarray`): Denoised images of
src/diffusers/pipelines/stable_diffusion/pipeline_output.py:32
↓ 4 callersClassGaussianFourierProjection
Gaussian Fourier embeddings for noise levels.
src/diffusers/models/embeddings.py:255
↓ 4 callersClassImageProjection
src/diffusers/models/embeddings.py:440
↓ 4 callersClassKandinsky3AttentionBlock
src/diffusers/models/unet_kandinsky3.py:500
next →1–100 of 1,211, ranked by callers