|
from typing import TYPE_CHECKING |
|
|
|
from ...utils import ( |
|
DIFFUSERS_SLOW_IMPORT, |
|
OptionalDependencyNotAvailable, |
|
_LazyModule, |
|
get_objects_from_module, |
|
is_flax_available, |
|
is_torch_available, |
|
is_transformers_available, |
|
) |
|
|
|
|
|
_dummy_objects = {} |
|
_import_structure = {} |
|
|
|
try: |
|
if not (is_transformers_available() and is_torch_available()): |
|
raise OptionalDependencyNotAvailable() |
|
except OptionalDependencyNotAvailable: |
|
from ...utils import dummy_torch_and_transformers_objects |
|
|
|
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects)) |
|
else: |
|
_import_structure["pipeline_pag_controlnet_sd"] = ["StableDiffusionControlNetPAGPipeline"] |
|
_import_structure["pipeline_pag_controlnet_sd_inpaint"] = ["StableDiffusionControlNetPAGInpaintPipeline"] |
|
_import_structure["pipeline_pag_controlnet_sd_xl"] = ["StableDiffusionXLControlNetPAGPipeline"] |
|
_import_structure["pipeline_pag_controlnet_sd_xl_img2img"] = ["StableDiffusionXLControlNetPAGImg2ImgPipeline"] |
|
_import_structure["pipeline_pag_hunyuandit"] = ["HunyuanDiTPAGPipeline"] |
|
_import_structure["pipeline_pag_kolors"] = ["KolorsPAGPipeline"] |
|
_import_structure["pipeline_pag_pixart_sigma"] = ["PixArtSigmaPAGPipeline"] |
|
_import_structure["pipeline_pag_sana"] = ["SanaPAGPipeline"] |
|
_import_structure["pipeline_pag_sd"] = ["StableDiffusionPAGPipeline"] |
|
_import_structure["pipeline_pag_sd_3"] = ["StableDiffusion3PAGPipeline"] |
|
_import_structure["pipeline_pag_sd_3_img2img"] = ["StableDiffusion3PAGImg2ImgPipeline"] |
|
_import_structure["pipeline_pag_sd_animatediff"] = ["AnimateDiffPAGPipeline"] |
|
_import_structure["pipeline_pag_sd_img2img"] = ["StableDiffusionPAGImg2ImgPipeline"] |
|
_import_structure["pipeline_pag_sd_inpaint"] = ["StableDiffusionPAGInpaintPipeline"] |
|
|
|
_import_structure["pipeline_pag_sd_xl"] = ["StableDiffusionXLPAGPipeline"] |
|
_import_structure["pipeline_pag_sd_xl_img2img"] = ["StableDiffusionXLPAGImg2ImgPipeline"] |
|
_import_structure["pipeline_pag_sd_xl_inpaint"] = ["StableDiffusionXLPAGInpaintPipeline"] |
|
|
|
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: |
|
try: |
|
if not (is_transformers_available() and is_torch_available()): |
|
raise OptionalDependencyNotAvailable() |
|
|
|
except OptionalDependencyNotAvailable: |
|
from ...utils.dummy_torch_and_transformers_objects import * |
|
else: |
|
from .pipeline_pag_controlnet_sd import StableDiffusionControlNetPAGPipeline |
|
from .pipeline_pag_controlnet_sd_inpaint import StableDiffusionControlNetPAGInpaintPipeline |
|
from .pipeline_pag_controlnet_sd_xl import StableDiffusionXLControlNetPAGPipeline |
|
from .pipeline_pag_controlnet_sd_xl_img2img import StableDiffusionXLControlNetPAGImg2ImgPipeline |
|
from .pipeline_pag_hunyuandit import HunyuanDiTPAGPipeline |
|
from .pipeline_pag_kolors import KolorsPAGPipeline |
|
from .pipeline_pag_pixart_sigma import PixArtSigmaPAGPipeline |
|
from .pipeline_pag_sana import SanaPAGPipeline |
|
from .pipeline_pag_sd import StableDiffusionPAGPipeline |
|
from .pipeline_pag_sd_3 import StableDiffusion3PAGPipeline |
|
from .pipeline_pag_sd_3_img2img import StableDiffusion3PAGImg2ImgPipeline |
|
from .pipeline_pag_sd_animatediff import AnimateDiffPAGPipeline |
|
from .pipeline_pag_sd_img2img import StableDiffusionPAGImg2ImgPipeline |
|
from .pipeline_pag_sd_inpaint import StableDiffusionPAGInpaintPipeline |
|
from .pipeline_pag_sd_xl import StableDiffusionXLPAGPipeline |
|
from .pipeline_pag_sd_xl_img2img import StableDiffusionXLPAGImg2ImgPipeline |
|
from .pipeline_pag_sd_xl_inpaint import StableDiffusionXLPAGInpaintPipeline |
|
|
|
else: |
|
import sys |
|
|
|
sys.modules[__name__] = _LazyModule( |
|
__name__, |
|
globals()["__file__"], |
|
_import_structure, |
|
module_spec=__spec__, |
|
) |
|
for name, value in _dummy_objects.items(): |
|
setattr(sys.modules[__name__], name, value) |
|
|
