o
    pi9                     @   sN  d Z ddlmZ ddlmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZ dgdgg g g g ddZze s:e W n eyU   dd	lmZ d
d eeD ed< Y n	w ed dg ze see W n ey   ddlmZ dd eeD ed< Y n'w ed g d g ded< ed g d ed g d dged< ze re se W n ey   ddlmZ dd eeD ed< Y n	w ed dg ze re se W n ey   ddlmZ d d eeD ed!< Y n
w ed d"d#g ze re se W n ey,   dd$lmZ d%d eeD ed&< Y n
w ed g d' ze rCe rCe sFe W n eyb   dd(lmZ d)d eeD ed*< Y n
w ed d+d,g ze rye rye s|e W n ey   dd-lmZ d.d eeD ed/< Y n
w ed g d0 ze re re se W n ey   dd1lmZ d2d eeD ed3< Y n
w ed g d4 ze re	 se W n ey    dd5lmZ d6d eeD ed7< Y n
w ed d8d9g ze re re
 se W n ey6   dd:lmZ d;d eeD ed<< Y n	w ed d=g z	e sGe W n eyc   dd>lmZ d?d eeD ed@< Y n&w dAgedB< dCgedD< dEgedF< dGgedH< ed dIg ed g dJ ze re se W n ey   ddKlmZ dLd eeD edM< Y n
w ed g dN z	e
 se W n ey   ddOlm Z  dPd ee D edQ< Y n	w ed dRg eserddSl!m"Z" z	e se W n ey   ddTl#T Y nw ddUl$m%Z% z	e se W n ey)   ddTl&T Y nw ddVl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQ ddlRmSZSmTZTmUZUmVZVmWZWmXZXmYZY ddWl$mZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZn ddlompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ ddXlmZ ze re se W n ey-   ddTlT Y nw ddYlomZ ze r=e s@e W n eyO   ddTlT Y n	w ddZlomZmZ ze rae sde W n eyt   ddTlT Y naw dd[l$mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm`Z`mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z# ze re re se W n ey   ddTl$T Y nw dd\l$m%Z%m&Z& ze re re se W n ey"   ddTl'T Y nw dd0l$m(Z(m)Z)m*Z* ze r@e r@e sCe W n eyS   ddTl+T Y nw dd4l$m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1 ze rye	 s|e W n ey   ddTl2T Y nw dd]l$m3Z3m4Z4 ze re re
 se W n ey   ddTl5T Y n	w dd^l$m6Z6 z	e se W n ey   ddTl7T Y nUw dd_l8m9Z9 dd`l:m;Z; ddal<m=Z= ddbl>m?Z? ddcl$m@Z@ ddJlomAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZI ze r8e s;e W n eyK   ddTlJT Y nw ddNl$mKZKmLZLmMZMmNZNmOZO z	e
 sle W n ey}   ddTlPT Y deS w dddl$mQZQ deS ddelRZReeSeT df eeUdge idheRjVeS< deS )iz0.30.0    )TYPE_CHECKING   )DIFFUSERS_SLOW_IMPORTOptionalDependencyNotAvailable_LazyModuleis_flax_availableis_k_diffusion_availableis_librosa_availableis_note_seq_availableis_onnx_availableis_scipy_availableis_sentencepiece_availableis_torch_availableis_torchsde_availableis_transformers_availableConfigMixinFromOriginalModelMixin)r   r   is_inflect_available is_invisible_watermark_availabler   is_k_diffusion_versionr	   r
   r   r   r   r   r   is_transformers_versionis_unidecode_availablelogging)configuration_utilsloadersmodels	pipelines
schedulersutils)dummy_onnx_objectsc                 C      g | ]	}| d s|qS _
startswith.0name r(   P/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/__init__.py
<listcomp>=   
    
r*   zutils.dummy_onnx_objectsr   OnnxRuntimeModel)dummy_pt_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*   J   s    zutils.dummy_pt_objectsr   )+AsymmetricAutoencoderKLAuraFlowTransformer2DModelAutoencoderKLAutoencoderKLCogVideoXAutoencoderKLTemporalDecoderAutoencoderOobleckAutoencoderTinyCogVideoXTransformer3DModelConsistencyDecoderVAEControlNetModelControlNetXSAdapterDiTTransformer2DModelFluxTransformer2DModelHunyuanDiT2DControlNetModelHunyuanDiT2DModel HunyuanDiT2DMultiControlNetModelI2VGenXLUNetKandinsky3UNetLatteTransformer3DModelLuminaNextDiT2DModel
ModelMixinMotionAdapterMultiAdapterPixArtTransformer2DModelPriorTransformerSD3ControlNetModelSD3MultiControlNetModelSD3Transformer2DModelSparseControlNetModelStableAudioDiTModelStableCascadeUNet
T2IAdapterT5FilmDecoderTransformer2DModelUNet1DModelUNet2DConditionModelUNet2DModelUNet3DConditionModelUNetControlNetXSModelUNetMotionModel UNetSpatioTemporalConditionModelUVit2DModelVQModel)get_constant_schedule!get_constant_schedule_with_warmupget_cosine_schedule_with_warmup2get_cosine_with_hard_restarts_schedule_with_warmupget_linear_schedule_with_warmup)get_polynomial_decay_schedule_with_warmupget_scheduleroptimization)AudioPipelineOutputAutoPipelineForImage2ImageAutoPipelineForInpaintingAutoPipelineForText2ImageConsistencyModelPipelineDanceDiffusionPipelineDDIMPipelineDDPMPipelineDiffusionPipelineDiTPipelineImagePipelineOutputKarrasVePipelineLDMPipelineLDMSuperResolutionPipelinePNDMPipelineRePaintPipelineScoreSdeVePipelineStableDiffusionMixinr   )#AmusedSchedulerCMStochasticIterativeSchedulerCogVideoXDDIMSchedulerCogVideoXDPMSchedulerDDIMInverseSchedulerDDIMParallelSchedulerDDIMSchedulerDDPMParallelSchedulerDDPMSchedulerDDPMWuerstchenSchedulerDEISMultistepScheduler"DPMSolverMultistepInverseSchedulerDPMSolverMultistepSchedulerDPMSolverSinglestepSchedulerEDMDPMSolverMultistepSchedulerEDMEulerSchedulerEulerAncestralDiscreteSchedulerEulerDiscreteSchedulerFlowMatchEulerDiscreteSchedulerFlowMatchHeunDiscreteSchedulerHeunDiscreteSchedulerIPNDMSchedulerKarrasVeSchedulerKDPM2AncestralDiscreteSchedulerKDPM2DiscreteSchedulerLCMSchedulerPNDMSchedulerRePaintSchedulerSASolverSchedulerSchedulerMixinScoreSdeVeSchedulerTCDSchedulerUnCLIPSchedulerUniPCMultistepSchedulerVQDiffusionSchedulerEMAModeltraining_utils)dummy_torch_and_scipy_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*      r+   z#utils.dummy_torch_and_scipy_objectsLMSDiscreteScheduler) dummy_torch_and_torchsde_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*      r+   z&utils.dummy_torch_and_torchsde_objects!CosineDPMSolverMultistepSchedulerDPMSolverSDEScheduler)$dummy_torch_and_transformers_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*      r+   z*utils.dummy_torch_and_transformers_objects)AltDiffusionImg2ImgPipelineAltDiffusionPipelineAmusedImg2ImgPipelineAmusedInpaintPipelineAmusedPipelineAnimateDiffControlNetPipelineAnimateDiffPAGPipelineAnimateDiffPipelineAnimateDiffSDXLPipeline#AnimateDiffSparseControlNetPipelineAnimateDiffVideoToVideoPipelineAudioLDM2PipelineAudioLDM2ProjectionModelAudioLDM2UNet2DConditionModelAudioLDMPipelineAuraFlowPipelineBlipDiffusionControlNetPipelineBlipDiffusionPipelineCLIPImageProjectionCogVideoXPipelineCycleDiffusionPipelineFluxPipelineHunyuanDiTControlNetPipelineHunyuanDiTPAGPipelineHunyuanDiTPipelineI2VGenXLPipelineIFImg2ImgPipeline IFImg2ImgSuperResolutionPipelineIFInpaintingPipeline#IFInpaintingSuperResolutionPipeline
IFPipelineIFSuperResolutionPipelineImageTextPipelineOutputKandinsky3Img2ImgPipelineKandinsky3PipelineKandinskyCombinedPipeline KandinskyImg2ImgCombinedPipelineKandinskyImg2ImgPipeline KandinskyInpaintCombinedPipelineKandinskyInpaintPipelineKandinskyPipelineKandinskyPriorPipelineKandinskyV22CombinedPipeline%KandinskyV22ControlnetImg2ImgPipelineKandinskyV22ControlnetPipeline#KandinskyV22Img2ImgCombinedPipelineKandinskyV22Img2ImgPipeline#KandinskyV22InpaintCombinedPipelineKandinskyV22InpaintPipelineKandinskyV22Pipeline KandinskyV22PriorEmb2EmbPipelineKandinskyV22PriorPipeline%LatentConsistencyModelImg2ImgPipelineLatentConsistencyModelPipelineLattePipelineLDMTextToImagePipelineLEditsPPPipelineStableDiffusion!LEditsPPPipelineStableDiffusionXLLuminaText2ImgPipelineMarigoldDepthPipelineMarigoldNormalsPipelineMusicLDMPipelinePaintByExamplePipelinePIAPipelinePixArtAlphaPipelinePixArtSigmaPAGPipelinePixArtSigmaPipelineSemanticStableDiffusionPipelineShapEImg2ImgPipelineShapEPipelineStableAudioPipelineStableAudioProjectionModelStableCascadeCombinedPipelineStableCascadeDecoderPipelineStableCascadePriorPipeline"StableDiffusion3ControlNetPipelineStableDiffusion3Img2ImgPipelineStableDiffusion3InpaintPipelineStableDiffusion3PAGPipelineStableDiffusion3PipelineStableDiffusionAdapterPipeline&StableDiffusionAttendAndExcitePipeline(StableDiffusionControlNetImg2ImgPipeline(StableDiffusionControlNetInpaintPipeline$StableDiffusionControlNetPAGPipeline!StableDiffusionControlNetPipeline#StableDiffusionControlNetXSPipeline StableDiffusionDepth2ImgPipelineStableDiffusionDiffEditPipelineStableDiffusionGLIGENPipeline&StableDiffusionGLIGENTextImagePipeline%StableDiffusionImageVariationPipelineStableDiffusionImg2ImgPipelineStableDiffusionInpaintPipeline$StableDiffusionInpaintPipelineLegacy&StableDiffusionInstructPix2PixPipeline$StableDiffusionLatentUpscalePipelineStableDiffusionLDM3DPipeline#StableDiffusionModelEditingPipelineStableDiffusionPAGPipelineStableDiffusionPanoramaPipeline StableDiffusionParadigmsPipelineStableDiffusionPipelineStableDiffusionPipelineSafe"StableDiffusionPix2PixZeroPipelineStableDiffusionSAGPipelineStableDiffusionUpscalePipeline StableDiffusionXLAdapterPipeline*StableDiffusionXLControlNetImg2ImgPipeline*StableDiffusionXLControlNetInpaintPipeline&StableDiffusionXLControlNetPAGPipeline#StableDiffusionXLControlNetPipeline%StableDiffusionXLControlNetXSPipeline StableDiffusionXLImg2ImgPipeline StableDiffusionXLInpaintPipeline(StableDiffusionXLInstructPix2PixPipeline#StableDiffusionXLPAGImg2ImgPipeline#StableDiffusionXLPAGInpaintPipelineStableDiffusionXLPAGPipelineStableDiffusionXLPipelineStableUnCLIPImg2ImgPipelineStableUnCLIPPipelineStableVideoDiffusionPipelineTextToVideoSDPipelineTextToVideoZeroPipelineTextToVideoZeroSDXLPipelineUnCLIPImageVariationPipelineUnCLIPPipelineUniDiffuserModelUniDiffuserPipelineUniDiffuserTextDecoder$VersatileDiffusionDualGuidedPipeline(VersatileDiffusionImageVariationPipelineVersatileDiffusionPipeline%VersatileDiffusionTextToImagePipelineVideoToVideoSDPipelineVQDiffusionPipelineWuerstchenCombinedPipelineWuerstchenDecoderPipelineWuerstchenPriorPipeline)4dummy_torch_and_transformers_and_k_diffusion_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   z:utils.dummy_torch_and_transformers_and_k_diffusion_objects!StableDiffusionKDiffusionPipeline#StableDiffusionXLKDiffusionPipeline)6dummy_torch_and_transformers_and_sentencepiece_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   z<utils.dummy_torch_and_transformers_and_sentencepiece_objects)KolorsImg2ImgPipelineKolorsPAGPipelineKolorsPipeline)-dummy_torch_and_transformers_and_onnx_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   z3utils.dummy_torch_and_transformers_and_onnx_objects)"OnnxStableDiffusionImg2ImgPipeline"OnnxStableDiffusionInpaintPipeline(OnnxStableDiffusionInpaintPipelineLegacyOnnxStableDiffusionPipeline"OnnxStableDiffusionUpscalePipelineStableDiffusionOnnxPipeline)dummy_torch_and_librosa_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   z%utils.dummy_torch_and_librosa_objectsAudioDiffusionPipelineMel)1dummy_transformers_and_torch_and_note_seq_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   z7utils.dummy_transformers_and_torch_and_note_seq_objectsSpectrogramDiffusionPipeline)dummy_flax_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   zutils.dummy_flax_objectsFlaxControlNetModelzmodels.controlnet_flaxFlaxModelMixinzmodels.modeling_flax_utilsFlaxUNet2DConditionModelz#models.unets.unet_2d_condition_flaxFlaxAutoencoderKLzmodels.vae_flaxFlaxDiffusionPipeline)	FlaxDDIMSchedulerFlaxDDPMSchedulerFlaxDPMSolverMultistepSchedulerFlaxEulerDiscreteSchedulerFlaxKarrasVeSchedulerFlaxLMSDiscreteSchedulerFlaxPNDMSchedulerFlaxSchedulerMixinFlaxScoreSdeVeScheduler)#dummy_flax_and_transformers_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   z)utils.dummy_flax_and_transformers_objects)%FlaxStableDiffusionControlNetPipeline"FlaxStableDiffusionImg2ImgPipeline"FlaxStableDiffusionInpaintPipelineFlaxStableDiffusionPipelineFlaxStableDiffusionXLPipeline)dummy_note_seq_objectsc                 C   r    r!   r#   r%   r(   r(   r)   r*     r+   zutils.dummy_note_seq_objectsMidiProcessor)r   )*)r,   )*r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   )ra   rb   rc   rd   r   r   r   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   )r   )r   )r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r  r	  r
  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  )r+  r,  )r9  r:  )r<  )r>  )r?  )r@  )rA  )rB  )rS  N__file____version__)module_specextra_objects(W  rV  typingr   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   _import_structurer   dirextendr-   r   r   r   r*  r-  r1  r8  r;  r=  rL  rR  r   r   utils.dummy_onnx_objectsr   r,   utils.dummy_pt_objectsr   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   r`   rY   rZ   r[   r\   r]   r^   r_   ra   rb   rc   rd   r   r   r   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   r   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   #utils.dummy_torch_and_scipy_objectsr   &utils.dummy_torch_and_torchsde_objectsr   r   *utils.dummy_torch_and_transformers_objectsr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r  r	  r
  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  :utils.dummy_torch_and_transformers_and_k_diffusion_objectsr+  r,  <utils.dummy_torch_and_transformers_and_sentencepiece_objectsr.  r/  r0  3utils.dummy_torch_and_transformers_and_onnx_objectsr2  r3  r4  r5  r6  r7  %utils.dummy_torch_and_librosa_objectsr9  r:  7utils.dummy_transformers_and_torch_and_note_seq_objectsr<  utils.dummy_flax_objectsmodels.controlnet_flaxr>  models.modeling_flax_utilsr?  #models.unets.unet_2d_condition_flaxr@  models.vae_flaxrA  rB  rC  rD  rE  rF  rG  rH  rI  rJ  rK  )utils.dummy_flax_and_transformers_objectsrM  rN  rO  rP  rQ  utils.dummy_note_seq_objectsrS  sys__name__globals__spec__modulesr(   r(   r(   r)   <module>   sZ   <0	
' 	
	


	
	$,\	%    8	P0
