

"""simple docstring"""




from ...configuration_utils import PretrainedConfig
from ...utils import logging


__A	:       Dict        =						logging.get_logger(__name__)

__A	:       Tuple        =						{
    'uw-madison/nystromformer-512': 'https://huggingface.co/uw-madison/nystromformer-512/resolve/main/config.json',
    # See all Nystromformer models at https://huggingface.co/models?filter=nystromformer
}





class      __UpperCamelCase					(  _A		):
		SCREAMING_SNAKE_CASE														=       "nystromformer"


		def __init__(self					:						Optional[Any]	,	__SCREAMING_SNAKE_CASE					:						Any=3_0_0_0_0	,	__SCREAMING_SNAKE_CASE					:						List[Any]=7_6_8	,	__SCREAMING_SNAKE_CASE					:						str=1_2	,	__SCREAMING_SNAKE_CASE					:						Dict=1_2	,	__SCREAMING_SNAKE_CASE					:						Optional[Any]=3_0_7_2	,	__SCREAMING_SNAKE_CASE					:						List[str]="gelu_new"	,	__SCREAMING_SNAKE_CASE					:						Optional[Any]=0.1	,	__SCREAMING_SNAKE_CASE					:						Tuple=0.1	,	__SCREAMING_SNAKE_CASE					:						Optional[Any]=5_1_0	,	__SCREAMING_SNAKE_CASE					:						str=2	,	__SCREAMING_SNAKE_CASE					:						str=6_4	,	__SCREAMING_SNAKE_CASE					:						List[Any]=6_4	,	__SCREAMING_SNAKE_CASE					:						List[str]=6_5	,	__SCREAMING_SNAKE_CASE					:						Optional[int]=False	,	__SCREAMING_SNAKE_CASE					:						Optional[Any]=0.0_2	,	__SCREAMING_SNAKE_CASE					:						List[Any]=1E-5	,	__SCREAMING_SNAKE_CASE					:						Any=1	,	__SCREAMING_SNAKE_CASE					:						List[Any]=0	,	__SCREAMING_SNAKE_CASE					:						Optional[Any]=2	,	**__SCREAMING_SNAKE_CASE					:						Any	,	):
							A     =   vocab_size
							A     =   max_position_embeddings
							A     =   hidden_size
							A     =   num_hidden_layers
							A     =   num_attention_heads
							A     =   intermediate_size
							A     =   hidden_act
							A     =   hidden_dropout_prob
							A     =   attention_probs_dropout_prob
							A     =   initializer_range
							A     =   type_vocab_size
							A     =   segment_means_seq_len
							A     =   num_landmarks
							A     =   conv_kernel_size
							A     =   inv_coeff_init_option
							A     =   layer_norm_eps
							super().__init__(pad_token_id=__SCREAMING_SNAKE_CASE	,	bos_token_id=__SCREAMING_SNAKE_CASE	,	eos_token_id=__SCREAMING_SNAKE_CASE	,	**__SCREAMING_SNAKE_CASE)







"""simple docstring"""







from ..utils import (
    OptionalDependencyNotAvailable,
    is_flax_available,
    is_invisible_watermark_available,
    is_k_diffusion_available,
    is_librosa_available,
    is_note_seq_available,
    is_onnx_available,
    is_torch_available,
    is_transformers_available,
)


try:
							if not is_torch_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_pt_objects import *  # noqa F403
else:
							from .consistency_models import ConsistencyModelPipeline
							from .dance_diffusion import DanceDiffusionPipeline
							from .ddim import DDIMPipeline
							from .ddpm import DDPMPipeline
							from .dit import DiTPipeline
							from .latent_diffusion import LDMSuperResolutionPipeline
							from .latent_diffusion_uncond import LDMPipeline
							from .pipeline_utils import AudioPipelineOutput, DiffusionPipeline, ImagePipelineOutput
							from .pndm import PNDMPipeline
							from .repaint import RePaintPipeline
							from .score_sde_ve import ScoreSdeVePipeline
							from .stochastic_karras_ve import KarrasVePipeline

try:
							if not (is_torch_available() and is_librosa_available()):
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_torch_and_librosa_objects import *  # noqa F403
else:
							from .audio_diffusion import AudioDiffusionPipeline, Mel

try:
							if not (is_torch_available() and is_transformers_available()):
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_torch_and_transformers_objects import *  # noqa F403
else:
							from .alt_diffusion import AltDiffusionImgaImgPipeline, AltDiffusionPipeline
							from .audioldm import AudioLDMPipeline
							from .controlnet import (
							    StableDiffusionControlNetImgaImgPipeline,
							    StableDiffusionControlNetInpaintPipeline,
							    StableDiffusionControlNetPipeline,
							)
							from .deepfloyd_if import (
							    IFImgaImgPipeline,
							    IFImgaImgSuperResolutionPipeline,
							    IFInpaintingPipeline,
							    IFInpaintingSuperResolutionPipeline,
							    IFPipeline,
							    IFSuperResolutionPipeline,
							)
							from .kandinsky import (
							    KandinskyImgaImgPipeline,
							    KandinskyInpaintPipeline,
							    KandinskyPipeline,
							    KandinskyPriorPipeline,
							)
							from .kandinskya_a import (
							    KandinskyVaaControlnetImgaImgPipeline,
							    KandinskyVaaControlnetPipeline,
							    KandinskyVaaImgaImgPipeline,
							    KandinskyVaaInpaintPipeline,
							    KandinskyVaaPipeline,
							    KandinskyVaaPriorEmbaEmbPipeline,
							    KandinskyVaaPriorPipeline,
							)
							from .latent_diffusion import LDMTextToImagePipeline
							from .paint_by_example import PaintByExamplePipeline
							from .semantic_stable_diffusion import SemanticStableDiffusionPipeline
							from .shap_e import ShapEImgaImgPipeline, ShapEPipeline
							from .stable_diffusion import (
							    CycleDiffusionPipeline,
							    StableDiffusionAttendAndExcitePipeline,
							    StableDiffusionDepthaImgPipeline,
							    StableDiffusionDiffEditPipeline,
							    StableDiffusionImageVariationPipeline,
							    StableDiffusionImgaImgPipeline,
							    StableDiffusionInpaintPipeline,
							    StableDiffusionInpaintPipelineLegacy,
							    StableDiffusionInstructPixaPixPipeline,
							    StableDiffusionLatentUpscalePipeline,
							    StableDiffusionLDMaDPipeline,
							    StableDiffusionModelEditingPipeline,
							    StableDiffusionPanoramaPipeline,
							    StableDiffusionParadigmsPipeline,
							    StableDiffusionPipeline,
							    StableDiffusionPixaPixZeroPipeline,
							    StableDiffusionSAGPipeline,
							    StableDiffusionUpscalePipeline,
							    StableUnCLIPImgaImgPipeline,
							    StableUnCLIPPipeline,
							)
							from .stable_diffusion_safe import StableDiffusionPipelineSafe
							from .text_to_video_synthesis import TextToVideoSDPipeline, TextToVideoZeroPipeline, VideoToVideoSDPipeline
							from .unclip import UnCLIPImageVariationPipeline, UnCLIPPipeline
							from .unidiffuser import ImageTextPipelineOutput, UniDiffuserModel, UniDiffuserPipeline, UniDiffuserTextDecoder
							from .versatile_diffusion import (
							    VersatileDiffusionDualGuidedPipeline,
							    VersatileDiffusionImageVariationPipeline,
							    VersatileDiffusionPipeline,
							    VersatileDiffusionTextToImagePipeline,
							)
							from .vq_diffusion import VQDiffusionPipeline


try:
							if not (is_torch_available() and is_transformers_available() and is_invisible_watermark_available()):
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_torch_and_transformers_and_invisible_watermark_objects import *  # noqa F403
else:
							from .stable_diffusion_xl import StableDiffusionXLImgaImgPipeline, StableDiffusionXLPipeline

try:
							if not is_onnx_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_onnx_objects import *  # noqa F403
else:
							from .onnx_utils import OnnxRuntimeModel

try:
							if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_torch_and_transformers_and_onnx_objects import *  # noqa F403
else:
							from .stable_diffusion import (
							    OnnxStableDiffusionImgaImgPipeline,
							    OnnxStableDiffusionInpaintPipeline,
							    OnnxStableDiffusionInpaintPipelineLegacy,
							    OnnxStableDiffusionPipeline,
							    OnnxStableDiffusionUpscalePipeline,
							    StableDiffusionOnnxPipeline,
							)

try:
							if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_torch_and_transformers_and_k_diffusion_objects import *  # noqa F403
else:
							from .stable_diffusion import StableDiffusionKDiffusionPipeline

try:
							if not is_flax_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_flax_objects import *  # noqa F403
else:
							from .pipeline_flax_utils import FlaxDiffusionPipeline


try:
							if not (is_flax_available() and is_transformers_available()):
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_flax_and_transformers_objects import *  # noqa F403
else:
							from .controlnet import FlaxStableDiffusionControlNetPipeline
							from .stable_diffusion import (
							    FlaxStableDiffusionImgaImgPipeline,
							    FlaxStableDiffusionInpaintPipeline,
							    FlaxStableDiffusionPipeline,
							)
try:
							if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							from ..utils.dummy_transformers_and_torch_and_note_seq_objects import *  # noqa F403
else:
							from .spectrogram_diffusion import MidiProcessor, SpectrogramDiffusionPipeline