Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LetterLiGo/SafeGen_CCS2024
/ types & classes
Types & classes
1,211 in github.com/LetterLiGo/SafeGen_CCS2024
⨍
Functions
7,383
◇
Types & classes
1,211
↳
Endpoints
4
↓ 123 callers
Class
OptionalDependencyNotAvailable
An error indicating that an optional dependency of Diffusers was not found in the environment.
src/diffusers/utils/import_utils.py:657
↓ 97 callers
Class
UNet2DConditionModel
src/diffusers/utils/dummy_pt_objects.py:230
↓ 86 callers
Class
AutoencoderKL
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 callers
Class
DDIMScheduler
src/diffusers/utils/dummy_pt_objects.py:693
↓ 75 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:46
↓ 61 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:584
↓ 59 callers
Class
StableDiffusionPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:950
↓ 47 callers
Class
PNDMScheduler
src/diffusers/utils/dummy_pt_objects.py:933
↓ 45 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:661
↓ 41 callers
Class
FrozenDict
src/diffusers/configuration_utils.py:50
↓ 37 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:111
↓ 32 callers
Class
TimestepEmbedding
src/diffusers/models/embeddings.py:190
↓ 32 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/transformer_2d.py:45
↓ 31 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:38
↓ 30 callers
Class
ConvResblock
scripts/convert_consistency_decoder.py:246
↓ 29 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
↓ 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:708
↓ 28 callers
Class
UNet2DModel
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 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:869
↓ 27 callers
Class
AttnProcessor
r""" Default processor for performing attention-related computations.
src/diffusers/models/attention_processor.py:696
↓ 27 callers
Class
DDPMScheduler
src/diffusers/utils/dummy_pt_objects.py:723
↓ 27 callers
Class
StableDiffusionXLPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1130
↓ 24 callers
Class
ResConvBlock
src/diffusers/models/unet_1d_blocks.py:375
↓ 21 callers
Class
Timesteps
src/diffusers/models/embeddings.py:238
↓ 20 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:105
↓ 20 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:172
↓ 19 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
↓ 18 callers
Class
EulerDiscreteScheduler
src/diffusers/utils/dummy_pt_objects.py:828
↓ 18 callers
Class
StableDiffusionXLImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1085
↓ 18 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/resnet.py:123
↓ 17 callers
Class
DecoderOutput
r""" Output of decoding method. Args: sample (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`):
src/diffusers/models/vae.py:34
↓ 17 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/resnet.py:216
↓ 15 callers
Class
Event
src/diffusers/pipelines/spectrogram_diffusion/midi_utils.py:116
↓ 15 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
↓ 14 callers
Class
StableDiffusionXLInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1100
↓ 14 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
↓ 13 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:50
↓ 12 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
↓ 12 callers
Class
SelfAttention1d
src/diffusers/models/unet_1d_blocks.py:317
↓ 11 callers
Class
UniDiffuserPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1280
↓ 10 callers
Class
DualTransformer2DModel
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 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/versatile_diffusion/modeling_text_unet.py:1400
↓ 10 callers
Class
StableDiffusionInpaintPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:830
↓ 10 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/transformer_temporal.py:41
↓ 9 callers
Class
ConsistencyModelPipeline
src/diffusers/utils/dummy_pt_objects.py:453
↓ 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:125
↓ 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:850
↓ 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:1206
↓ 8 callers
Class
StableDiffusionImg2ImgPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:815
↓ 8 callers
Class
StableDiffusionPix2PixZeroPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:980
↓ 8 callers
Class
UnCLIPScheduler
src/diffusers/utils/dummy_pt_objects.py:993
↓ 8 callers
Class
WuerstchenLayerNorm
src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py:24
↓ 7 callers
Class
DDPMPipeline
src/diffusers/utils/dummy_pt_objects.py:498
↓ 7 callers
Class
PriorTransformer
src/diffusers/utils/dummy_pt_objects.py:155
↓ 7 callers
Class
StableDiffusionPanoramaPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:920
↓ 7 callers
Class
StableDiffusionXLAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1025
↓ 7 callers
Class
StableDiffusionXLWatermarker
src/diffusers/pipelines/stable_diffusion_xl/watermark.py:17
↓ 7 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/unet_2d_blocks.py:505
↓ 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:26
↓ 6 callers
Class
AttnProcsLayers
src/diffusers/loaders/utils.py:20
↓ 6 callers
Class
CMStochasticIterativeScheduler
src/diffusers/utils/dummy_pt_objects.py:648
↓ 6 callers
Class
CustomOutput
tests/others/test_outputs.py:14
↓ 6 callers
Class
DPMSolverMultistepScheduler
src/diffusers/utils/dummy_pt_objects.py:783
↓ 6 callers
Class
DiagonalGaussianDistribution
src/diffusers/models/vae.py:765
↓ 6 callers
Class
EventRange
src/diffusers/pipelines/spectrogram_diffusion/midi_utils.py:109
↓ 6 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:493
↓ 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
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/versatile_diffusion/modeling_text_unet.py:1381
↓ 6 callers
Class
SampleObject
tests/others/test_config.py:32
↓ 6 callers
Class
StableDiffusionAdapterPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:665
↓ 6 callers
Class
T5LayerNorm
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 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:1042
↓ 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
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:1168
↓ 5 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:96
↓ 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/resnet.py:80
↓ 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/vae.py:46
↓ 5 callers
Class
EulerAncestralDiscreteScheduler
src/diffusers/utils/dummy_pt_objects.py:813
↓ 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
ResnetDownsampleBlock2D
src/diffusers/models/unet_2d_blocks.py:1610
↓ 5 callers
Class
ResnetUpsampleBlock2D
src/diffusers/models/unet_2d_blocks.py:2861
↓ 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:1665
↓ 5 callers
Class
StableDiffusionInpaintPipelineLegacy
src/diffusers/utils/dummy_torch_and_transformers_objects.py:845
↓ 5 callers
Class
StableDiffusionInstructPix2PixPipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:860
↓ 5 callers
Class
StableDiffusionUpscalePipeline
src/diffusers/utils/dummy_torch_and_transformers_objects.py:1010
↓ 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:113
↓ 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:933
↓ 4 callers
Class
AutoencoderKLOutput
Output of AutoencoderKL encoding method. Args: latent_dist (`DiagonalGaussianDistribution`): Encoded outputs of `Encoder
src/diffusers/models/modeling_outputs.py:7
↓ 4 callers
Class
CustomDiffusionAttnProcessor
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 callers
Class
DDPMWuerstchenScheduler
src/diffusers/utils/dummy_pt_objects.py:738
↓ 4 callers
Class
EMAModel
Exponential Moving Average of models weights
src/diffusers/training_utils.py:76
↓ 4 callers
Class
FlaxStableDiffusionPipelineOutput
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 callers
Class
GaussianFourierProjection
Gaussian Fourier embeddings for noise levels.
src/diffusers/models/embeddings.py:255
↓ 4 callers
Class
ImageProjection
src/diffusers/models/embeddings.py:440
↓ 4 callers
Class
Kandinsky3AttentionBlock
src/diffusers/models/unet_kandinsky3.py:500
next →
1–100 of 1,211, ranked by callers