o
    Gi[
                     @   s  d dl mZ ddlmZmZmZmZmZmZ i Z	i Z
ddgiZze r&e s)e W n ey@   ddlmZ e	ee Y nw dged< z#e rhd d	lZd d
lmZ G dd deZee
d< eedshee_W n	 eyr   Y nw eswerze r~e se W n ey   ddlT Y d	S w ddlmZ ddlmZ d	S d d	lZeee d eedeje< e	 D ]\ZZ e!eje ee  qe
 D ]\ZZ e!eje ee  qd	S )    )TYPE_CHECKING   )DIFFUSERS_SLOW_IMPORTOptionalDependencyNotAvailable_LazyModuleget_objects_from_moduleis_torch_availableis_transformers_availablepipeline_outputPRXPipelineOutput)$dummy_torch_and_transformers_objectsPRXPipelinepipeline_prxN)T5GemmaEncoderc                       s    e Zd Ze fddZ  ZS )r   c                    sL   d|vrddl m} ||}t|dr|j|d< t j|g|R i |S )Nconfigr   )T5GemmaConfigencoder)1transformers.models.t5gemma.configuration_t5gemmar   from_pretrainedhasattrr   super)clspretrained_model_name_or_pathargskwargsr   r   	__class__ T/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/pipelines/prx/__init__.pyr   #   s   


zT5GemmaEncoder.from_pretrained)__name__
__module____qualname__classmethodr   __classcell__r   r   r   r   r   "   s    r   )*   )r   )r   __file__)module_spec)"typingr   utilsr   r   r   r   r   r	   _dummy_objects_additional_imports_import_structurer   updatetransformers,transformers.models.t5gemma.modeling_t5gemmar   _T5GemmaEncoderr   ImportError*utils.dummy_torch_and_transformers_objectsr
   r   r   r   sysr   globals__spec__modulesitemsnamevaluesetattrr   r   r   r   <module>   s`     



