
    +h                        S SK JrJr   " S S\S9r " S S\S9r " S S\S9r " S	 S
\S9r " S S\S9r " S S\S9r " S S\S9r	 " S S\S9r
 " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9rg)    )DummyObjectrequires_backendsc                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxControlNetModel   flaxc                     [        U S/5        g Nr   r   selfargskwargss      \/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/utils/dummy_flax_objects.py__init__FlaxControlNetModel.__init__       $)    c                     [        U S/5        g r
   r   clsr   r   s      r   from_configFlaxControlNetModel.from_config       #x(r   c                     [        U S/5        g r
   r   r   s      r   from_pretrained#FlaxControlNetModel.from_pretrained   r   r    N
__name__
__module____qualname____firstlineno__	_backendsr   classmethodr   r   __static_attributes__r   r   r   r   r      6    I* ) ) ) )r   r   )	metaclassc                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxModelMixin   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxModelMixin.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxModelMixin.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxModelMixin.from_pretrained   r   r   r   Nr   r   r   r   r*   r*      r'   r   r*   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxUNet2DConditionModel#   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxUNet2DConditionModel.__init__&   r   r   c                     [        U S/5        g r
   r   r   s      r   r   $FlaxUNet2DConditionModel.from_config)   r   r   c                     [        U S/5        g r
   r   r   s      r   r   (FlaxUNet2DConditionModel.from_pretrained-   r   r   r   Nr   r   r   r   r3   r3   #   r'   r   r3   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxAutoencoderKL2   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxAutoencoderKL.__init__5   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxAutoencoderKL.from_config8   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxAutoencoderKL.from_pretrained<   r   r   r   Nr   r   r   r   r<   r<   2   r'   r   r<   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxDiffusionPipelineA   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxDiffusionPipeline.__init__D   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxDiffusionPipeline.from_configG   r   r   c                     [        U S/5        g r
   r   r   s      r   r   %FlaxDiffusionPipeline.from_pretrainedK   r   r   r   Nr   r   r   r   rE   rE   A   r'   r   rE   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxDDIMSchedulerP   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxDDIMScheduler.__init__S   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxDDIMScheduler.from_configV   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxDDIMScheduler.from_pretrainedZ   r   r   r   Nr   r   r   r   rN   rN   P   r'   r   rN   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxDDPMScheduler_   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxDDPMScheduler.__init__b   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxDDPMScheduler.from_confige   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxDDPMScheduler.from_pretrainedi   r   r   r   Nr   r   r   r   rW   rW   _   r'   r   rW   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxDPMSolverMultistepSchedulern   r   c                     [        U S/5        g r
   r   r   s      r   r   (FlaxDPMSolverMultistepScheduler.__init__q   r   r   c                     [        U S/5        g r
   r   r   s      r   r   +FlaxDPMSolverMultistepScheduler.from_configt   r   r   c                     [        U S/5        g r
   r   r   s      r   r   /FlaxDPMSolverMultistepScheduler.from_pretrainedx   r   r   r   Nr   r   r   r   r`   r`   n   r'   r   r`   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxEulerDiscreteScheduler}   r   c                     [        U S/5        g r
   r   r   s      r   r   #FlaxEulerDiscreteScheduler.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &FlaxEulerDiscreteScheduler.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   *FlaxEulerDiscreteScheduler.from_pretrained   r   r   r   Nr   r   r   r   ri   ri   }   r'   r   ri   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxKarrasVeScheduler   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxKarrasVeScheduler.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxKarrasVeScheduler.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   %FlaxKarrasVeScheduler.from_pretrained   r   r   r   Nr   r   r   r   rr   rr      r'   r   rr   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxLMSDiscreteScheduler   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxLMSDiscreteScheduler.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   $FlaxLMSDiscreteScheduler.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   (FlaxLMSDiscreteScheduler.from_pretrained   r   r   r   Nr   r   r   r   r{   r{      r'   r   r{   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxPNDMScheduler   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxPNDMScheduler.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxPNDMScheduler.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlaxPNDMScheduler.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxSchedulerMixin   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxSchedulerMixin.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FlaxSchedulerMixin.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   "FlaxSchedulerMixin.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlaxScoreSdeVeScheduler   r   c                     [        U S/5        g r
   r   r   s      r   r    FlaxScoreSdeVeScheduler.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   #FlaxScoreSdeVeScheduler.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   'FlaxScoreSdeVeScheduler.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   N)utilsr   r   r   r*   r3   r<   rE   rN   rW   r`   ri   rr   r{   r   r   r   r   r   r   <module>r      s    2)K )){ )) ))+ ))k ))+ ))+ )) )); ))k )) ))+ )); )) )r   