declare-lab/tango2
92
1a
2 ��'d� � @ s$ d Z ddlmZ ddlmZmZmZmZmZm Z m3Z4m Z m Z m
Z
mZmZmZmZ ze � s^e� �W n eyz ddlT Y n0 ddlmZ ze
� s�e� �W n ey� ddlT Y n�0 ddlmZmZmZmZmZm Z m Z m Z m Z m Z m!Z! ddl"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z) ddlm*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6 dd l7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZK dd5lLmMZM ze
� �r�e � �s�e� �W n e�y� ddlNT Y n0 dd l7mOZO ze
� �re� �se� �W n e�y& ddlPT Y n�0 dd lQmRZR dd
lmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZs z"e
� �r�e� �r�e� �s�e� �W n e�y� ddltT Y n0 ddlmuZu z"e
� �r&e� �r&e � �s,e� �W n e�yJ ddlvT Y n"0 ddlmwZwmxZxmyZymzZzm{Z{m|Z| ze
� �r~e � �s�e� �W n e�y� ddl}T Y n0 ddlm~Z~mZ z"e� �r�e
� �r�e6� �s�e� �W n e�y� ddl�T Y n0 ddlm�Z� ze� �se� �W n e�y. ddl�T Y nf0 ddl�m�Z� ddl�m�Z� ddl�m�Z� ddl�m�Z� ddlm�Z� ddl7m�Z�m�Z�m�Z�m�Z�m�Z�m�Z�m�Z�m�Z� ze� �r�e� �s�e� �W n e�y� ddl�T Y n0 ddlm�Z�m�Z�m�Z�m�Z� ze7� �s�e� �W n e�y ddl�T Y n0 ddlm�Z� dS )z 0.15.0.dev0� )� ConfigMixin)� OptionalDependencyNotAvailable�is_flax_available�is_inflect_available�is_k_diffusion_available�is_k_diffusion_version�is_librosa_available�is_note_seq_available�is_onnx_available�is_scipy_available�is_torch_available�is_transformers_available�is_transformers_version�is_unidecode_available�logging)�*)�OnnxRuntimeModel) �
AutoencoderKL�ControlNetModel�8ModelMixin�PriorTransformer�
T5FilmDecoder�Transformer2DModel� UNet1DModel�UNet2DConditionModel� UNet2DModel�UNet3DConditionModel�VQModel)�get_constant_schedule�!get_constant_schedule_with_warmup� get_cosine_schedule_with_warmup�2get_cosine_with_hard_restarts_schedule_with_warmup� get_linear_schedule_with_warmup�)get_polynomial_decay_schedule_with_warmup�
get_scheduler)
�AudioPipelineOutput�DanceDiffusionPipeline� DDIMPipeline� DDPMPipeline�DiffusionPipeline� DiTPipeline�ImagePipelineOutput�KarrasVePipeline� LDMPipeline�LDMSuperResolutionPipeline� PNDMPipeline�RePaintPipeline�ScoreSdeVePipeline)�DDIMInverseScheduler�
DDIMScheduler�
DDPMScheduler�DEISMultistepScheduler�DPMSolverMultistepScheduler� DPMSolverSinglestepScheduler� EulerAncestralDiscreteScheduler�EulerDiscreteScheduler�HeunDiscreteScheduler�IPNDMScheduler�KarrasVeScheduler� KDPM2AncestralDiscreteScheduler�KDPM2DiscreteScheduler�
PNDMScheduler�RePaintScheduler�SchedulerMixin�ScoreSdeVeScheduler�UnCLIPScheduler�UniPCMultistepScheduler�VQDiffusionScheduler)�EMAModel)�LMSDiscreteScheduler)�TextualInversionLoaderMixin)!�AltDiffusionImg2ImgPipeline�AltDiffusionPipeline�AudioLDMPipeline�CycleDiffusionPipeline�LDMTextToImagePipeline�PaintByExamplePipeline� SemanticStableDiffusionPipeline�&StableDiffusionAttendAndExcitePipeline�!StableDiffusionControlNetPipeline� StableDiffusionDepth2ImgPipeline�%StableDiffusionImageVariationPipeline� StableDiffusionImg2ImgPipeline� StableDiffusionInpaintPipeline�$StableDiffusionInpaintPipelineLegacy�&StableDiffusionInstructPix2PixPipeline�$StableDiffusionLatentUpscalePipeline�#StableDiffusionModelEditingPipeline� StableDiffusionPanoramaPipeline�StableDiffusionPipeline�StableDiffusionPipelineSafe�"StableDiffusionPix2PixZeroPipeline�StableDiffusionSAGPipeline� StableDiffusionUpscalePipeline�StableUnCLIPImg2ImgPipeline�StableUnCLIPPipeline�TextToVideoSDPipeline� UnCLIPImageVariationPipeline�UnCLIPPipeline�$VersatileDiffusionDualGuidedPipeline�(VersatileDiffusionImageVariationPipeline�VersatileDiffusionPipeline�%VersatileDiffusionTextToImagePipeline�VQDiffusionPipeline)�!StableDiffusionKDiffusionPipeline)�"OnnxStableDiffusionImg2ImgPipeline�"OnnxStableDiffusionInpaintPipeline�(OnnxStableDiffusionInpaintPipelineLegacy�OnnxStableDiffusionPipeline�"OnnxStableDiffusionUpscalePipeline�StableDiffusionOnnxPipeline)�AudioDiffusionPipeline�Mel)� SpectrogramDiffusionPipeline)�FlaxControlNetModel)�FlaxModelMixin)�FlaxUNet2DConditionModel)�FlaxAutoencoderKL)�FlaxDiffusionPipeline)�FlaxDDIMScheduler�FlaxDDPMScheduler� FlaxDPMSolverMultistepScheduler�FlaxKarrasVeScheduler�FlaxLMSDiscreteScheduler�FlaxPNDMScheduler�FlaxSchedulerMixin�FlaxScoreSdeVeScheduler)�%FlaxStableDiffusionControlNetPipeline�"FlaxStableDiffusionImg2ImgPipeline�"FlaxStableDiffusionInpaintPipeline�FlaxStableDiffusionPipeline)�
MidiProcessorN)�� __version__Zconfiguration_utilsr �utilsr r r r r r r r9 r r r
r r r Zutils.dummy_onnx_objectsZ pipelinesr Zutils.dummy_pt_objects�modelsr r r r r r r r r r r � optimizationr r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 Z10schedulersr2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rB rC rD rE Ztraining_utilsrF Z#utils.dummy_torch_and_scipy_objectsrG Z*utils.dummy_torch_and_transformers_objects�loadersrH rI rJ rK rL rM rN rO rP rQ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri Z:utils.dummy_torch_and_transformers_and_k_diffusion_objectsrj Z3utils.dummy_torch_and_transformers_and_onnx_objectsrk rl rm rn ro rp Z%utils.dummy_torch_and_librosa_objectsrq rr Z7utils.dummy_transformers_and_torch_and_note_seq_objectsrs Zutils.dummy_flax_objectsZmodels.controlnet_flaxrt Zmodels.modeling_flax_utilsru Z models.unet_2d_condition_flaxrv Zmodels.vae_flaxrw rx ry rz r{ r| r} r~ r r� Z)utils.dummy_flax_and_transformers_objectsr� r� r� r� Z utils.dummy_note_seq_objectsr� � r� r� �G/home/deep/Projects/audio_diffusion/diffusers/src/diffusers/__init__.py�<module> s� @11 12 4
$ <X 13 14 �$15 16 1718 19 ( 2021