Compare commits

...

1 Commits

Author SHA1 Message Date
Dhruv Nair
90cca148de fix auto pipeline lazy import 2023-09-14 09:31:03 +00:00

View File

@@ -17,42 +17,62 @@ import inspect
from collections import OrderedDict from collections import OrderedDict
from ..configuration_utils import ConfigMixin from ..configuration_utils import ConfigMixin
from ..utils import DIFFUSERS_CACHE from ..utils import (
from .controlnet import ( DIFFUSERS_CACHE,
StableDiffusionControlNetImg2ImgPipeline, OptionalDependencyNotAvailable,
StableDiffusionControlNetInpaintPipeline, is_torch_available,
StableDiffusionControlNetPipeline, is_transformers_available,
StableDiffusionXLControlNetImg2ImgPipeline,
StableDiffusionXLControlNetPipeline,
) )
from .deepfloyd_if import IFImg2ImgPipeline, IFInpaintingPipeline, IFPipeline
from .kandinsky import (
KandinskyCombinedPipeline, try:
KandinskyImg2ImgCombinedPipeline, if not (is_torch_available() and is_transformers_available()):
KandinskyImg2ImgPipeline, raise OptionalDependencyNotAvailable()
KandinskyInpaintCombinedPipeline, except OptionalDependencyNotAvailable:
KandinskyInpaintPipeline, from ..utils.dummy_torch_and_transformers_objects import * # noqa F403
KandinskyPipeline, else:
) from .controlnet import (
from .kandinsky2_2 import ( StableDiffusionControlNetImg2ImgPipeline,
KandinskyV22CombinedPipeline, StableDiffusionControlNetInpaintPipeline,
KandinskyV22Img2ImgCombinedPipeline, StableDiffusionControlNetPipeline,
KandinskyV22Img2ImgPipeline, StableDiffusionXLControlNetImg2ImgPipeline,
KandinskyV22InpaintCombinedPipeline, StableDiffusionXLControlNetPipeline,
KandinskyV22InpaintPipeline, )
KandinskyV22Pipeline, from .deepfloyd_if import (
) IFImg2ImgPipeline,
from .stable_diffusion import ( IFInpaintingPipeline,
StableDiffusionImg2ImgPipeline, IFPipeline,
StableDiffusionInpaintPipeline, )
StableDiffusionPipeline, from .kandinsky import (
) KandinskyCombinedPipeline,
from .stable_diffusion_xl import ( KandinskyImg2ImgCombinedPipeline,
StableDiffusionXLImg2ImgPipeline, KandinskyImg2ImgPipeline,
StableDiffusionXLInpaintPipeline, KandinskyInpaintCombinedPipeline,
StableDiffusionXLPipeline, KandinskyInpaintPipeline,
) KandinskyPipeline,
from .wuerstchen import WuerstchenCombinedPipeline, WuerstchenDecoderPipeline )
from .kandinsky2_2 import (
KandinskyV22CombinedPipeline,
KandinskyV22Img2ImgCombinedPipeline,
KandinskyV22Img2ImgPipeline,
KandinskyV22InpaintCombinedPipeline,
KandinskyV22InpaintPipeline,
KandinskyV22Pipeline,
)
from .stable_diffusion import (
StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline,
StableDiffusionPipeline,
)
from .stable_diffusion_xl import (
StableDiffusionXLImg2ImgPipeline,
StableDiffusionXLInpaintPipeline,
StableDiffusionXLPipeline,
)
from .wuerstchen import (
WuerstchenCombinedPipeline,
WuerstchenDecoderPipeline,
)
AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(