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