o
    ۷i                     @   s  d dl mZmZ G dd dedZG dd dedZG dd dedZG d	d
 d
edZG dd dedZG dd dedZG dd dedZ	G dd dedZ
G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZdS )    )DummyObjectrequires_backendsc                   @   2   e Zd ZdgZdd Zedd Zedd ZdS )	FlaxControlNetModelflaxc                 O      t | dg d S Nr   r   selfargskwargs r   X/home/ubuntu/vllm_env/lib/python3.10/site-packages/diffusers/utils/dummy_flax_objects.py__init__      zFlaxControlNetModel.__init__c                 O   r   r   r	   clsr   r   r   r   r   from_config      zFlaxControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   from_pretrained   r   z#FlaxControlNetModel.from_pretrainedN__name__
__module____qualname__	_backendsr   classmethodr   r   r   r   r   r   r          
r   )	metaclassc                   @   r   )	FlaxModelMixinr   c                 O   r   r   r	   r
   r   r   r   r      r   zFlaxModelMixin.__init__c                 O   r   r   r	   r   r   r   r   r      r   zFlaxModelMixin.from_configc                 O   r   r   r	   r   r   r   r   r      r   zFlaxModelMixin.from_pretrainedNr   r   r   r   r   r      r   r   c                   @   r   )	FlaxUNet2DConditionModelr   c                 O   r   r   r	   r
   r   r   r   r   &   r   z!FlaxUNet2DConditionModel.__init__c                 O   r   r   r	   r   r   r   r   r   )   r   z$FlaxUNet2DConditionModel.from_configc                 O   r   r   r	   r   r   r   r   r   -   r   z(FlaxUNet2DConditionModel.from_pretrainedNr   r   r   r   r   r    #   r   r    c                   @   r   )	FlaxAutoencoderKLr   c                 O   r   r   r	   r
   r   r   r   r   5   r   zFlaxAutoencoderKL.__init__c                 O   r   r   r	   r   r   r   r   r   8   r   zFlaxAutoencoderKL.from_configc                 O   r   r   r	   r   r   r   r   r   <   r   z!FlaxAutoencoderKL.from_pretrainedNr   r   r   r   r   r!   2   r   r!   c                   @   r   )	FlaxDiffusionPipeliner   c                 O   r   r   r	   r
   r   r   r   r   D   r   zFlaxDiffusionPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   G   r   z!FlaxDiffusionPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   K   r   z%FlaxDiffusionPipeline.from_pretrainedNr   r   r   r   r   r"   A   r   r"   c                   @   r   )	FlaxDDIMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   S   r   zFlaxDDIMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   V   r   zFlaxDDIMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   Z   r   z!FlaxDDIMScheduler.from_pretrainedNr   r   r   r   r   r#   P   r   r#   c                   @   r   )	FlaxDDPMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   b   r   zFlaxDDPMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   e   r   zFlaxDDPMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   i   r   z!FlaxDDPMScheduler.from_pretrainedNr   r   r   r   r   r$   _   r   r$   c                   @   r   )	FlaxDPMSolverMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   q   r   z(FlaxDPMSolverMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   t   r   z+FlaxDPMSolverMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   x   r   z/FlaxDPMSolverMultistepScheduler.from_pretrainedNr   r   r   r   r   r%   n   r   r%   c                   @   r   )	FlaxEulerDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r      r   z#FlaxEulerDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r      r   z&FlaxEulerDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r      r   z*FlaxEulerDiscreteScheduler.from_pretrainedNr   r   r   r   r   r&   }   r   r&   c                   @   r   )	FlaxKarrasVeSchedulerr   c                 O   r   r   r	   r
   r   r   r   r      r   zFlaxKarrasVeScheduler.__init__c                 O   r   r   r	   r   r   r   r   r      r   z!FlaxKarrasVeScheduler.from_configc                 O   r   r   r	   r   r   r   r   r      r   z%FlaxKarrasVeScheduler.from_pretrainedNr   r   r   r   r   r'      r   r'   c                   @   r   )	FlaxLMSDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r      r   z!FlaxLMSDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r      r   z$FlaxLMSDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r      r   z(FlaxLMSDiscreteScheduler.from_pretrainedNr   r   r   r   r   r(      r   r(   c                   @   r   )	FlaxPNDMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r      r   zFlaxPNDMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r      r   zFlaxPNDMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r      r   z!FlaxPNDMScheduler.from_pretrainedNr   r   r   r   r   r)      r   r)   c                   @   r   )	FlaxSchedulerMixinr   c                 O   r   r   r	   r
   r   r   r   r      r   zFlaxSchedulerMixin.__init__c                 O   r   r   r	   r   r   r   r   r      r   zFlaxSchedulerMixin.from_configc                 O   r   r   r	   r   r   r   r   r      r   z"FlaxSchedulerMixin.from_pretrainedNr   r   r   r   r   r*      r   r*   c                   @   r   )	FlaxScoreSdeVeSchedulerr   c                 O   r   r   r	   r
   r   r   r   r      r   z FlaxScoreSdeVeScheduler.__init__c                 O   r   r   r	   r   r   r   r   r      r   z#FlaxScoreSdeVeScheduler.from_configc                 O   r   r   r	   r   r   r   r   r      r   z'FlaxScoreSdeVeScheduler.from_pretrainedNr   r   r   r   r   r+      r   r+   N)utilsr   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r   r   r   r   <module>   s   