Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/YesianRohn/TextSSR
/ types & classes
Types & classes
1,980 in github.com/YesianRohn/TextSSR
⨍
Functions
13,352
◇
Types & classes
1,980
↳
Endpoints
6
↓ 190 callers
Class
OptionalDependencyNotAvailable
An error indicating that an optional dependency of Diffusers was not found in the environment.
diffusers/src/diffusers/utils/import_utils.py:784
↓ 171 callers
Class
VaeImageProcessor
Image processor for VAE. Args: do_resize (`bool`, *optional*, defaults to `True`): Whether to downscale the image's (hei
diffusers/src/diffusers/image_processor.py:60
↓ 120 callers
Class
AutoencoderKL
diffusers/src/diffusers/utils/dummy_pt_objects.py:35
↓ 113 callers
Class
UNet2DConditionModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:545
↓ 104 callers
Class
FrozenDict
diffusers/src/diffusers/configuration_utils.py:55
↓ 91 callers
Class
DDIMScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1098
↓ 81 callers
Class
StableDiffusionPipelineOutput
Output class for Stable Diffusion pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL
diffusers/src/diffusers/pipelines/stable_diffusion/pipeline_output.py:11
↓ 73 callers
Class
_LazyModule
Module class that surfaces all objects but only performs associated imports when the objects are requested.
diffusers/src/diffusers/utils/import_utils.py:790
↓ 61 callers
Class
ResnetBlock2D
r""" A Resnet block. Parameters: in_channels (`int`): The number of channels in the input. out_channels (`int`, *optional*, d
diffusers/src/diffusers/models/resnet.py:189
↓ 58 callers
Class
StableDiffusionPipeline
r""" Pipeline for text-to-image generation using Stable Diffusion. This model inherits from [`DiffusionPipeline`]. Check the superclass docum
diffusers/examples/community/sd_text2img_k_diffusion.py:44
↓ 50 callers
Class
Attention
r""" A cross attention layer. Parameters: query_dim (`int`): The number of channels in the query. cross_attention
diffusers/src/diffusers/models/attention_processor.py:41
↓ 47 callers
Class
ImagePipelineOutput
Output class for image pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL images of
diffusers/src/diffusers/pipelines/pipeline_utils.py:104
↓ 47 callers
Class
TimestepEmbedding
diffusers/src/diffusers/models/embeddings.py:717
↓ 38 callers
Class
PNDMScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1398
↓ 37 callers
Class
AttnProcessor
r""" Default processor for performing attention-related computations.
diffusers/src/diffusers/models/attention_processor.py:715
↓ 36 callers
Class
StableDiffusionXLPipelineOutput
Output class for Stable Diffusion pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL
diffusers/src/diffusers/pipelines/stable_diffusion_xl/pipeline_output.py:11
↓ 33 callers
Class
DDPMScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1128
↓ 33 callers
Class
Timesteps
diffusers/src/diffusers/models/embeddings.py:765
↓ 31 callers
Class
CaptureLogger
Args: Context manager to capture `logging` streams logger: 'logging` logger object Returns: The captured output is availa
diffusers/src/diffusers/utils/testing_utils.py:863
↓ 30 callers
Class
AttnAddedKVProcessor
r""" Processor for performing attention-related computations with extra learnable key and value matrices for the text encoder.
diffusers/src/diffusers/models/attention_processor.py:891
↓ 30 callers
Class
ConvResblock
diffusers/scripts/convert_consistency_decoder.py:246
↓ 27 callers
Class
ControlNetModel
A ControlNet model. Args: in_channels (`int`, defaults to 4): The number of channels in the input sample. flip_s
diffusers/src/diffusers/models/controlnet.py:111
↓ 27 callers
Class
EulerDiscreteScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1263
↓ 26 callers
Class
MultiControlNetModel
r""" Multiple `ControlNetModel` wrapper class for Multi-ControlNet This module is a wrapper for multiple instances of the `ControlNetModel`.
diffusers/src/diffusers/pipelines/controlnet/multicontrolnet.py:15
↓ 25 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
diffusers/src/diffusers/models/unets/unet_2d.py:40
↓ 24 callers
Class
ResConvBlock
diffusers/src/diffusers/models/unets/unet_1d_blocks.py:375
↓ 24 callers
Class
Transformer2DModel
A 2D Transformer model for image-like data. Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads
diffusers/src/diffusers/models/transformers/transformer_2d.py:39
↓ 22 callers
Class
Downsample2D
A 2D downsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and outp
diffusers/src/diffusers/models/downsampling.py:69
↓ 22 callers
Class
Upsample2D
A 2D upsampling layer with an optional convolution. Parameters: channels (`int`): number of channels in the inputs and output
diffusers/src/diffusers/models/upsampling.py:76
↓ 22 callers
Class
VideoProcessor
r"""Simple video processor.
diffusers/src/diffusers/video_processor.py:25
↓ 21 callers
Class
StableDiffusionXLImg2ImgPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1835
↓ 21 callers
Class
VQModel
diffusers/src/diffusers/models/vq_model.py:25
↓ 20 callers
Class
FeedForward
r""" A feed-forward layer. Parameters: dim (`int`): The number of channels in the input. dim_out (`int`, *optional*): The num
diffusers/src/diffusers/models/attention.py:1144
↓ 20 callers
Class
RMSNorm
diffusers/src/diffusers/models/normalization.py:409
↓ 20 callers
Class
StableDiffusionXLWatermarker
diffusers/src/diffusers/pipelines/stable_diffusion_xl/watermark.py:17
↓ 19 callers
Class
DecoderOutput
r""" Output of decoding method. Args: sample (`torch.Tensor` of shape `(batch_size, num_channels, height, width)`): The d
diffusers/src/diffusers/models/autoencoders/vae.py:34
↓ 19 callers
Class
EMAModel
Exponential Moving Average of models weights
diffusers/src/diffusers/training_utils.py:276
↓ 17 callers
Class
StableDiffusionXLPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1925
↓ 16 callers
Class
FlowMatchEulerDiscreteScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1278
↓ 16 callers
Class
SchedulerOutput
Base class for the output of a scheduler's `step` function. Args: prev_sample (`torch.Tensor` of shape `(batch_size, num_channels, h
diffusers/src/diffusers/schedulers/scheduling_utils.py:61
↓ 16 callers
Class
StableDiffusionXLInpaintPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1850
↓ 15 callers
Class
Event
diffusers/src/diffusers/pipelines/deprecated/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
diffusers/src/diffusers/models/adapter.py:217
↓ 14 callers
Class
AttnProcessor2_0
r""" Processor for implementing scaled dot-product attention (enabled by default if you're using PyTorch 2.0).
diffusers/src/diffusers/models/attention_processor.py:2296
↓ 14 callers
Class
BasicTransformerBlock
r""" A basic Transformer block. Parameters: dim (`int`): The number of channels in the input and output. num_attention_heads
diffusers/src/diffusers/models/attention.py:213
↓ 13 callers
Class
FusedAttnProcessor2_0
r""" Processor for implementing scaled dot-product attention (enabled by default if you're using PyTorch 2.0). It uses fused projection layers
diffusers/src/diffusers/models/attention_processor.py:3065
↓ 12 callers
Class
FP32LayerNorm
diffusers/src/diffusers/models/normalization.py:88
↓ 12 callers
Class
ResnetBlockCondNorm2D
r""" A Resnet block that use normalization layer that incorporate conditioning information. Parameters: in_channels (`int`): The numb
diffusers/src/diffusers/models/resnet.py:44
↓ 12 callers
Class
SelfAttention1d
diffusers/src/diffusers/models/unets/unet_1d_blocks.py:317
↓ 12 callers
Class
StableDiffusionImg2ImgPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1490
↓ 12 callers
Class
StableDiffusionInpaintPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1505
↓ 12 callers
Class
Transformer2DModelOutput
diffusers/src/diffusers/models/transformers/transformer_2d.py:32
↓ 11 callers
Class
AnimateDiffPipelineOutput
r""" Output class for AnimateDiff pipelines. Args: frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
diffusers/src/diffusers/pipelines/animatediff/pipeline_output.py:12
↓ 11 callers
Class
ConsistencyModelPipeline
diffusers/src/diffusers/utils/dummy_pt_objects.py:798
↓ 11 callers
Class
DDPMPipeline
diffusers/src/diffusers/utils/dummy_pt_objects.py:843
↓ 11 callers
Class
DDPMWuerstchenScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1143
↓ 11 callers
Class
MotionAdapter
diffusers/src/diffusers/utils/dummy_pt_objects.py:350
↓ 11 callers
Class
UniDiffuserPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:2075
↓ 10 callers
Class
AudioPipelineOutput
Output class for audio pipelines. Args: audios (`np.ndarray`) List of denoised audio samples of a NumPy array of shape `
diffusers/src/diffusers/pipelines/pipeline_utils.py:118
↓ 10 callers
Class
DualTransformer2DModel
Dual transformer wrapper that combines two `Transformer2DModel`s for mixed inference. Parameters: num_attention_heads (`int`, *optio
diffusers/src/diffusers/models/transformers/dual_transformer_2d.py:22
↓ 10 callers
Class
FlaxResnetBlock2D
Flax implementation of 2D Resnet Block. Args: in_channels (`int`): Input channels out_channels (`int`):
diffusers/src/diffusers/models/vae_flax.py:125
↓ 10 callers
Class
ResnetBlockFlat
diffusers/src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1446
↓ 10 callers
Class
Transformer2DModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:515
↓ 9 callers
Class
CogVideoXCausalConv3d
r"""A 3D causal convolution layer that pads the input tensor to ensure causality in CogVideoX Model. Args: in_channels (`int`): Number of
diffusers/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py:69
↓ 9 callers
Class
DPMSolverMultistepScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1188
↓ 9 callers
Class
PatchEmbed
2D Image to Patch Embedding with support for SD3 cropping.
diffusers/src/diffusers/models/embeddings.py:183
↓ 8 callers
Class
AdaLayerNormZero
r""" Norm layer adaptive layer norm zero (adaLN-Zero). Parameters: embedding_dim (`int`): The size of each embedding vector.
diffusers/src/diffusers/models/normalization.py:100
↓ 8 callers
Class
AudioLDM2Pipeline
r""" Pipeline for text-to-audio generation using AudioLDM2. This model inherits from [`DiffusionPipeline`]. Check the superclass documentatio
diffusers/src/diffusers/pipelines/audioldm2/pipeline_audioldm2.py:136
↓ 8 callers
Class
AudioLDMPipeline
r""" Pipeline for text-to-audio generation using AudioLDM. This model inherits from [`DiffusionPipeline`]. Check the superclass documentation
diffusers/src/diffusers/pipelines/audioldm/pipeline_audioldm.py:52
↓ 8 callers
Class
CMStochasticIterativeScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1023
↓ 8 callers
Class
MusicLDMPipeline
r""" Pipeline for text-to-audio generation using MusicLDM. This model inherits from [`DiffusionPipeline`]. Check the superclass documentation
diffusers/src/diffusers/pipelines/musicldm/pipeline_musicldm.py:67
↓ 8 callers
Class
PixArtAlphaTextProjection
Projects caption embeddings. Also handles dropout for classifier-free guidance. Adapted from https://github.com/PixArt-alpha/PixArt-alpha/bl
diffusers/src/diffusers/models/embeddings.py:1497
↓ 8 callers
Class
ResidualTemporalBlock1D
Residual 1D block with temporal convolutions. Parameters: inp_channels (`int`): Number of input channels. out_channels (`int
diffusers/src/diffusers/models/resnet.py:424
↓ 8 callers
Class
SDCascadeLayerNorm
diffusers/src/diffusers/models/unets/unet_stable_cascade.py:31
↓ 8 callers
Class
SpatioTemporalResBlock
r""" A SpatioTemporal Resnet block. Parameters: in_channels (`int`): The number of channels in the input. out_channels (`int`
diffusers/src/diffusers/models/resnet.py:635
↓ 8 callers
Class
StableAudioPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1145
↓ 8 callers
Class
StableCascadeUNet
diffusers/src/diffusers/models/unets/unet_stable_cascade.py:137
↓ 8 callers
Class
UnCLIPScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1488
↓ 8 callers
Class
WuerstchenLayerNorm
diffusers/src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py:7
↓ 7 callers
Class
AdaLayerNorm
r""" Norm layer modified to incorporate timestep embeddings. Parameters: embedding_dim (`int`): The size of each embedding vector.
diffusers/src/diffusers/models/normalization.py:31
↓ 7 callers
Class
AdaLayerNormContinuous
diffusers/src/diffusers/models/normalization.py:277
↓ 7 callers
Class
AttnProcsLayers
diffusers/src/diffusers/loaders/utils.py:20
↓ 7 callers
Class
DiagonalGaussianDistribution
diffusers/src/diffusers/models/autoencoders/vae.py:767
↓ 7 callers
Class
FluxPipelineOutput
Output class for Stable Diffusion pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL
diffusers/src/diffusers/pipelines/flux/pipeline_output.py:11
↓ 7 callers
Class
FluxTransformer2DModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:215
↓ 7 callers
Class
PriorTransformer
diffusers/src/diffusers/utils/dummy_pt_objects.py:395
↓ 7 callers
Class
SD3Transformer2DModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:440
↓ 7 callers
Class
StableDiffusion3PipelineOutput
Output class for Stable Diffusion pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL
diffusers/src/diffusers/pipelines/stable_diffusion_3/pipeline_output.py:11
↓ 7 callers
Class
StableDiffusionPanoramaPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1625
↓ 7 callers
Class
StableDiffusionXLAdapterPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1730
↓ 7 callers
Class
UNetMidBlock2D
A 2D UNet mid-block [`UNetMidBlock2D`] with multiple residual blocks and optional attention blocks. Args: in_channels (`int`): The n
diffusers/src/diffusers/models/unets/unet_2d_blocks.py:589
↓ 7 callers
Class
UNetMidBlock2DCrossAttn
diffusers/src/diffusers/models/unets/unet_2d_blocks.py:744
↓ 6 callers
Class
AnimateDiffTransformer3D
A Transformer model for video-like data. Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads to
diffusers/src/diffusers/models/unets/unet_motion_model.py:65
↓ 6 callers
Class
AutoencoderTiny
diffusers/src/diffusers/utils/dummy_pt_objects.py:95
↓ 6 callers
Class
CustomOutput
diffusers/tests/others/test_outputs.py:14
↓ 6 callers
Class
EventRange
diffusers/src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:109
↓ 6 callers
Class
IFPipelineOutput
r""" Output class for Stable Diffusion pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`): List of denoise
diffusers/src/diffusers/pipelines/deepfloyd_if/pipeline_output.py:11
↓ 6 callers
Class
ImagePipelineOutput
diffusers/src/diffusers/utils/dummy_pt_objects.py:888
↓ 6 callers
Class
KandinskyPriorPipelineOutput
Output class for KandinskyPriorPipeline. Args: image_embeds (`torch.Tensor`) clip image embeddings for text prompt
diffusers/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py:113
↓ 6 callers
Class
LinearMultiDim
diffusers/src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1427
next →
1–100 of 1,980, ranked by callers