MCPcopy Create free account

hub / github.com/YesianRohn/TextSSR / types & classes

Types & classes1,980 in github.com/YesianRohn/TextSSR

↓ 190 callersClassOptionalDependencyNotAvailable
An error indicating that an optional dependency of Diffusers was not found in the environment.
diffusers/src/diffusers/utils/import_utils.py:784
↓ 171 callersClassVaeImageProcessor
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 callersClassAutoencoderKL
diffusers/src/diffusers/utils/dummy_pt_objects.py:35
↓ 113 callersClassUNet2DConditionModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:545
↓ 104 callersClassFrozenDict
diffusers/src/diffusers/configuration_utils.py:55
↓ 91 callersClassDDIMScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1098
↓ 81 callersClassStableDiffusionPipelineOutput
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 callersClass_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 callersClassResnetBlock2D
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 callersClassStableDiffusionPipeline
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 callersClassAttention
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 callersClassImagePipelineOutput
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 callersClassTimestepEmbedding
diffusers/src/diffusers/models/embeddings.py:717
↓ 38 callersClassPNDMScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1398
↓ 37 callersClassAttnProcessor
r""" Default processor for performing attention-related computations.
diffusers/src/diffusers/models/attention_processor.py:715
↓ 36 callersClassStableDiffusionXLPipelineOutput
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 callersClassDDPMScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1128
↓ 33 callersClassTimesteps
diffusers/src/diffusers/models/embeddings.py:765
↓ 31 callersClassCaptureLogger
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 callersClassAttnAddedKVProcessor
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 callersClassConvResblock
diffusers/scripts/convert_consistency_decoder.py:246
↓ 27 callersClassControlNetModel
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 callersClassEulerDiscreteScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1263
↓ 26 callersClassMultiControlNetModel
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 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
diffusers/src/diffusers/models/unets/unet_2d.py:40
↓ 24 callersClassResConvBlock
diffusers/src/diffusers/models/unets/unet_1d_blocks.py:375
↓ 24 callersClassTransformer2DModel
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 callersClassDownsample2D
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 callersClassUpsample2D
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 callersClassVideoProcessor
r"""Simple video processor.
diffusers/src/diffusers/video_processor.py:25
↓ 21 callersClassStableDiffusionXLImg2ImgPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1835
↓ 21 callersClassVQModel
diffusers/src/diffusers/models/vq_model.py:25
↓ 20 callersClassFeedForward
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 callersClassRMSNorm
diffusers/src/diffusers/models/normalization.py:409
↓ 20 callersClassStableDiffusionXLWatermarker
diffusers/src/diffusers/pipelines/stable_diffusion_xl/watermark.py:17
↓ 19 callersClassDecoderOutput
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 callersClassEMAModel
Exponential Moving Average of models weights
diffusers/src/diffusers/training_utils.py:276
↓ 17 callersClassStableDiffusionXLPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1925
↓ 16 callersClassFlowMatchEulerDiscreteScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1278
↓ 16 callersClassSchedulerOutput
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 callersClassStableDiffusionXLInpaintPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1850
↓ 15 callersClassEvent
diffusers/src/diffusers/pipelines/deprecated/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
diffusers/src/diffusers/models/adapter.py:217
↓ 14 callersClassAttnProcessor2_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 callersClassBasicTransformerBlock
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 callersClassFusedAttnProcessor2_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 callersClassFP32LayerNorm
diffusers/src/diffusers/models/normalization.py:88
↓ 12 callersClassResnetBlockCondNorm2D
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 callersClassSelfAttention1d
diffusers/src/diffusers/models/unets/unet_1d_blocks.py:317
↓ 12 callersClassStableDiffusionImg2ImgPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1490
↓ 12 callersClassStableDiffusionInpaintPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1505
↓ 12 callersClassTransformer2DModelOutput
diffusers/src/diffusers/models/transformers/transformer_2d.py:32
↓ 11 callersClassAnimateDiffPipelineOutput
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 callersClassConsistencyModelPipeline
diffusers/src/diffusers/utils/dummy_pt_objects.py:798
↓ 11 callersClassDDPMPipeline
diffusers/src/diffusers/utils/dummy_pt_objects.py:843
↓ 11 callersClassDDPMWuerstchenScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1143
↓ 11 callersClassMotionAdapter
diffusers/src/diffusers/utils/dummy_pt_objects.py:350
↓ 11 callersClassUniDiffuserPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:2075
↓ 10 callersClassAudioPipelineOutput
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 callersClassDualTransformer2DModel
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 callersClassFlaxResnetBlock2D
Flax implementation of 2D Resnet Block. Args: in_channels (`int`): Input channels out_channels (`int`):
diffusers/src/diffusers/models/vae_flax.py:125
↓ 10 callersClassResnetBlockFlat
diffusers/src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1446
↓ 10 callersClassTransformer2DModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:515
↓ 9 callersClassCogVideoXCausalConv3d
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 callersClassDPMSolverMultistepScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1188
↓ 9 callersClassPatchEmbed
2D Image to Patch Embedding with support for SD3 cropping.
diffusers/src/diffusers/models/embeddings.py:183
↓ 8 callersClassAdaLayerNormZero
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 callersClassAudioLDM2Pipeline
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 callersClassAudioLDMPipeline
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 callersClassCMStochasticIterativeScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1023
↓ 8 callersClassMusicLDMPipeline
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 callersClassPixArtAlphaTextProjection
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 callersClassResidualTemporalBlock1D
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 callersClassSDCascadeLayerNorm
diffusers/src/diffusers/models/unets/unet_stable_cascade.py:31
↓ 8 callersClassSpatioTemporalResBlock
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 callersClassStableAudioPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1145
↓ 8 callersClassStableCascadeUNet
diffusers/src/diffusers/models/unets/unet_stable_cascade.py:137
↓ 8 callersClassUnCLIPScheduler
diffusers/src/diffusers/utils/dummy_pt_objects.py:1488
↓ 8 callersClassWuerstchenLayerNorm
diffusers/src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py:7
↓ 7 callersClassAdaLayerNorm
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 callersClassAdaLayerNormContinuous
diffusers/src/diffusers/models/normalization.py:277
↓ 7 callersClassAttnProcsLayers
diffusers/src/diffusers/loaders/utils.py:20
↓ 7 callersClassDiagonalGaussianDistribution
diffusers/src/diffusers/models/autoencoders/vae.py:767
↓ 7 callersClassFluxPipelineOutput
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 callersClassFluxTransformer2DModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:215
↓ 7 callersClassPriorTransformer
diffusers/src/diffusers/utils/dummy_pt_objects.py:395
↓ 7 callersClassSD3Transformer2DModel
diffusers/src/diffusers/utils/dummy_pt_objects.py:440
↓ 7 callersClassStableDiffusion3PipelineOutput
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 callersClassStableDiffusionPanoramaPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1625
↓ 7 callersClassStableDiffusionXLAdapterPipeline
diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py:1730
↓ 7 callersClassUNetMidBlock2D
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 callersClassUNetMidBlock2DCrossAttn
diffusers/src/diffusers/models/unets/unet_2d_blocks.py:744
↓ 6 callersClassAnimateDiffTransformer3D
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 callersClassAutoencoderTiny
diffusers/src/diffusers/utils/dummy_pt_objects.py:95
↓ 6 callersClassCustomOutput
diffusers/tests/others/test_outputs.py:14
↓ 6 callersClassEventRange
diffusers/src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:109
↓ 6 callersClassIFPipelineOutput
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 callersClassImagePipelineOutput
diffusers/src/diffusers/utils/dummy_pt_objects.py:888
↓ 6 callersClassKandinskyPriorPipelineOutput
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 callersClassLinearMultiDim
diffusers/src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py:1427
next →1–100 of 1,980, ranked by callers