o
    [5i                     @   s@  d dl mZ 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G d d! d!edZG d"d# d#edZd$S )%    )ClassVar)DummyObjectrequires_backendsc                   @   0   e Zd ZU dgZeed< dd Zedd ZdS )ORTDiffusionPipeline	diffusers	_backendsc                 O      t | dg d S Nr   r   selfargskwargs r   o/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/optimum/onnxruntime/dummy_objects.py__init__      zORTDiffusionPipeline.__init__c                 O   r	   r
   r   clsr   r   r   r   r   from_pretrained      z$ORTDiffusionPipeline.from_pretrainedN	__name__
__module____qualname__r   r   __annotations__r   classmethodr   r   r   r   r   r      
   
 r   )	metaclassc                   @   r   )ORTPipelineForText2Imager   r   c                 O   r	   r
   r   r   r   r   r   r   "   r   z!ORTPipelineForText2Image.__init__c                 O   r	   r
   r   r   r   r   r   r   %   r   z(ORTPipelineForText2Image.from_pretrainedNr   r   r   r   r   r       r   r    c                   @   r   )ORTPipelineForImage2Imager   r   c                 O   r	   r
   r   r   r   r   r   r   -   r   z"ORTPipelineForImage2Image.__init__c                 O   r	   r
   r   r   r   r   r   r   0   r   z)ORTPipelineForImage2Image.from_pretrainedNr   r   r   r   r   r!   *   r   r!   c                   @   r   )ORTPipelineForInpaintingr   r   c                 O   r	   r
   r   r   r   r   r   r   8   r   z!ORTPipelineForInpainting.__init__c                 O   r	   r
   r   r   r   r   r   r   ;   r   z(ORTPipelineForInpainting.from_pretrainedNr   r   r   r   r   r"   5   r   r"   c                   @   r   )ORTStableDiffusionPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r   C   r   z#ORTStableDiffusionPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r   F   r   z*ORTStableDiffusionPipeline.from_pretrainedNr   r   r   r   r   r#   @   r   r#   c                   @   r   )!ORTStableDiffusionImg2ImgPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r   N   r   z*ORTStableDiffusionImg2ImgPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r   Q   r   z1ORTStableDiffusionImg2ImgPipeline.from_pretrainedNr   r   r   r   r   r$   K   r   r$   c                   @   r   )!ORTStableDiffusionInpaintPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r   Y   r   z*ORTStableDiffusionInpaintPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r   \   r   z1ORTStableDiffusionInpaintPipeline.from_pretrainedNr   r   r   r   r   r%   V   r   r%   c                   @   r   )ORTStableDiffusionXLPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r   d   r   z%ORTStableDiffusionXLPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r   g   r   z,ORTStableDiffusionXLPipeline.from_pretrainedNr   r   r   r   r   r&   a   r   r&   c                   @   r   )#ORTStableDiffusionXLImg2ImgPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r   o   r   z,ORTStableDiffusionXLImg2ImgPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r   r   r   z3ORTStableDiffusionXLImg2ImgPipeline.from_pretrainedNr   r   r   r   r   r'   l   r   r'   c                   @   r   )#ORTStableDiffusionXLInpaintPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r   z   r   z,ORTStableDiffusionXLInpaintPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r   }   r   z3ORTStableDiffusionXLInpaintPipeline.from_pretrainedNr   r   r   r   r   r(   w   r   r(   c                   @   r   )!ORTLatentConsistencyModelPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r      r   z*ORTLatentConsistencyModelPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r      r   z1ORTLatentConsistencyModelPipeline.from_pretrainedNr   r   r   r   r   r)      r   r)   c                   @   r   )(ORTLatentConsistencyModelImg2ImgPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r      r   z1ORTLatentConsistencyModelImg2ImgPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r      r   z8ORTLatentConsistencyModelImg2ImgPipeline.from_pretrainedNr   r   r   r   r   r*      r   r*   c                   @   r   )ORTStableDiffusion3Pipeliner   r   c                 O   r	   r
   r   r   r   r   r   r      r   z$ORTStableDiffusion3Pipeline.__init__c                 O   r	   r
   r   r   r   r   r   r      r   z+ORTStableDiffusion3Pipeline.from_pretrainedNr   r   r   r   r   r+      r   r+   c                   @   r   )"ORTStableDiffusion3Img2ImgPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r      r   z+ORTStableDiffusion3Img2ImgPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r      r   z2ORTStableDiffusion3Img2ImgPipeline.from_pretrainedNr   r   r   r   r   r,      r   r,   c                   @   r   )"ORTStableDiffusion3InpaintPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r      r   z+ORTStableDiffusion3InpaintPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r      r   z2ORTStableDiffusion3InpaintPipeline.from_pretrainedNr   r   r   r   r   r-      r   r-   c                   @   r   )ORTFluxPipeliner   r   c                 O   r	   r
   r   r   r   r   r   r      r   zORTFluxPipeline.__init__c                 O   r	   r
   r   r   r   r   r   r      r   zORTFluxPipeline.from_pretrainedNr   r   r   r   r   r.      r   r.   N)typingr   optimum.utils.import_utilsr   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r   r   r   r   <module>   s$   