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	dg d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< d	ge
d
< dge
d< dge
d< dge
d< dge
d< dge
d< dge
d< dge
d< esqerze rxe s{e W n ey   ddlT Y d%S w ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd!lmZ dd"lmZ dd#lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%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/Z0e1e)j-e* e/e0 qd%S )(    )TYPE_CHECKING   )DIFFUSERS_SLOW_IMPORTOptionalDependencyNotAvailable_LazyModuleget_objects_from_moduleis_torch_availableis_transformers_available	timesteps)fast27_timestepssmart100_timestepssmart185_timestepssmart27_timestepssmart50_timestepssuper100_timestepssuper27_timestepssuper40_timesteps)$dummy_torch_and_transformers_objects
IFPipelinepipeline_ifIFImg2ImgPipelinepipeline_if_img2img IFImg2ImgSuperResolutionPipeline#pipeline_if_img2img_superresolutionIFInpaintingPipelinepipeline_if_inpainting#IFInpaintingSuperResolutionPipeline&pipeline_if_inpainting_superresolutionIFSuperResolutionPipelinepipeline_if_superresolutionIFPipelineOutputpipeline_outputIFSafetyCheckersafety_checkerIFWatermarker	watermark)*   )r   )r   )r   )r   )r   )r   )r    )r"   )r   r   r   r   r   r   r   r   )r$   N__file__)module_spec)2typingr   utilsr   r   r   r   r   r	   _dummy_objects_import_structurer   update*utils.dummy_torch_and_transformers_objectsr   r   r   r   r   r   r   r   r   r   r   r   r!   r    r#   r"   r
   r   r   r   r   r   r   r   r   r%   r$   sys__name__globals__spec__modulesitemsnamevaluesetattr r9   r9   ]/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/pipelines/deepfloyd_if/__init__.py<module>   sd     









(
