
    1;ji                        d dl mZmZ  G d de          Z G d de          Z G d de          Z G d	 d
e          Z G d de          Z G d de          Z G d de          Z	 G d de          Z
 G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          ZdS )    )DummyObjectrequires_backendsc                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxControlNetModelflaxc                 (    t          | dg           d S Nr   r   selfargskwargss      \/root/voice-cloning/.venv/lib/python3.11/site-packages/diffusers/utils/dummy_flax_objects.py__init__zFlaxControlNetModel.__init__       $)))))    c                 (    t          | dg           d S r	   r
   clsr   r   s      r   from_configzFlaxControlNetModel.from_config       #x(((((r   c                 (    t          | dg           d S r	   r
   r   s      r   from_pretrainedz#FlaxControlNetModel.from_pretrained   r   r   N__name__
__module____qualname__	_backendsr   classmethodr   r    r   r   r   r      `        I* * * ) ) [) ) ) [) ) )r   r   )	metaclassc                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxModelMixinr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxModelMixin.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxModelMixin.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxModelMixin.from_pretrained   r   r   Nr   r    r   r   r$   r$      r!   r   r$   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxUNet2DConditionModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxUNet2DConditionModel.__init__&   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$FlaxUNet2DConditionModel.from_config)   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(FlaxUNet2DConditionModel.from_pretrained-   r   r   Nr   r    r   r   r)   r)   #   r!   r   r)   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxAutoencoderKLr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxAutoencoderKL.__init__5   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxAutoencoderKL.from_config8   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxAutoencoderKL.from_pretrained<   r   r   Nr   r    r   r   r.   r.   2   r!   r   r.   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxDiffusionPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxDiffusionPipeline.__init__D   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxDiffusionPipeline.from_configG   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%FlaxDiffusionPipeline.from_pretrainedK   r   r   Nr   r    r   r   r3   r3   A   r!   r   r3   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxDDIMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxDDIMScheduler.__init__S   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxDDIMScheduler.from_configV   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxDDIMScheduler.from_pretrainedZ   r   r   Nr   r    r   r   r8   r8   P   r!   r   r8   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxDDPMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxDDPMScheduler.__init__b   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxDDPMScheduler.from_confige   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxDDPMScheduler.from_pretrainedi   r   r   Nr   r    r   r   r=   r=   _   r!   r   r=   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxDPMSolverMultistepSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(FlaxDPMSolverMultistepScheduler.__init__q   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z+FlaxDPMSolverMultistepScheduler.from_configt   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z/FlaxDPMSolverMultistepScheduler.from_pretrainedx   r   r   Nr   r    r   r   rB   rB   n   r!   r   rB   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxEulerDiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#FlaxEulerDiscreteScheduler.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z&FlaxEulerDiscreteScheduler.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z*FlaxEulerDiscreteScheduler.from_pretrained   r   r   Nr   r    r   r   rG   rG   }   r!   r   rG   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxKarrasVeSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxKarrasVeScheduler.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxKarrasVeScheduler.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%FlaxKarrasVeScheduler.from_pretrained   r   r   Nr   r    r   r   rL   rL      r!   r   rL   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxLMSDiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxLMSDiscreteScheduler.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$FlaxLMSDiscreteScheduler.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(FlaxLMSDiscreteScheduler.from_pretrained   r   r   Nr   r    r   r   rQ   rQ      r!   r   rQ   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxPNDMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxPNDMScheduler.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxPNDMScheduler.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!FlaxPNDMScheduler.from_pretrained   r   r   Nr   r    r   r   rV   rV      r!   r   rV   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxSchedulerMixinr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxSchedulerMixin.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zFlaxSchedulerMixin.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"FlaxSchedulerMixin.from_pretrained   r   r   Nr   r    r   r   r[   r[      r!   r   r[   c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlaxScoreSdeVeSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z FlaxScoreSdeVeScheduler.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#FlaxScoreSdeVeScheduler.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z'FlaxScoreSdeVeScheduler.from_pretrained   r   r   Nr   r    r   r   r`   r`      r!   r   r`   N)utilsr   r   r   r$   r)   r.   r3   r8   r=   rB   rG   rL   rQ   rV   r[   r`   r    r   r   <module>re      s   2 2 2 2 2 2 2 2) ) ) ) )K ) ) ) )) ) ) ) ){ ) ) ) )) ) ) ) ) ) ) ) )) ) ) ) )+ ) ) ) )) ) ) ) )k ) ) ) )) ) ) ) )+ ) ) ) )) ) ) ) )+ ) ) ) )) ) ) ) ) ) ) ) )) ) ) ) ); ) ) ) )) ) ) ) )k ) ) ) )) ) ) ) ) ) ) ) )) ) ) ) )+ ) ) ) )) ) ) ) ); ) ) ) )) ) ) ) ) ) ) ) ) ) )r   