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,
OptionalDependencyNotAvailable,
is_torch_available,
is_transformers_available,
)
try:
if not (is_torch_available() and is_transformers_available()):
raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
from ..utils.dummy_torch_and_transformers_objects import * # noqa F403
else:
from .controlnet import (
StableDiffusionControlNetImg2ImgPipeline, StableDiffusionControlNetImg2ImgPipeline,
StableDiffusionControlNetInpaintPipeline, StableDiffusionControlNetInpaintPipeline,
StableDiffusionControlNetPipeline, StableDiffusionControlNetPipeline,
StableDiffusionXLControlNetImg2ImgPipeline, StableDiffusionXLControlNetImg2ImgPipeline,
StableDiffusionXLControlNetPipeline, StableDiffusionXLControlNetPipeline,
) )
from .deepfloyd_if import IFImg2ImgPipeline, IFInpaintingPipeline, IFPipeline from .deepfloyd_if import (
from .kandinsky import ( IFImg2ImgPipeline,
IFInpaintingPipeline,
IFPipeline,
)
from .kandinsky import (
KandinskyCombinedPipeline, KandinskyCombinedPipeline,
KandinskyImg2ImgCombinedPipeline, KandinskyImg2ImgCombinedPipeline,
KandinskyImg2ImgPipeline, KandinskyImg2ImgPipeline,
KandinskyInpaintCombinedPipeline, KandinskyInpaintCombinedPipeline,
KandinskyInpaintPipeline, KandinskyInpaintPipeline,
KandinskyPipeline, KandinskyPipeline,
) )
from .kandinsky2_2 import ( from .kandinsky2_2 import (
KandinskyV22CombinedPipeline, KandinskyV22CombinedPipeline,
KandinskyV22Img2ImgCombinedPipeline, KandinskyV22Img2ImgCombinedPipeline,
KandinskyV22Img2ImgPipeline, KandinskyV22Img2ImgPipeline,
KandinskyV22InpaintCombinedPipeline, KandinskyV22InpaintCombinedPipeline,
KandinskyV22InpaintPipeline, KandinskyV22InpaintPipeline,
KandinskyV22Pipeline, KandinskyV22Pipeline,
) )
from .stable_diffusion import ( from .stable_diffusion import (
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline, StableDiffusionInpaintPipeline,
StableDiffusionPipeline, StableDiffusionPipeline,
) )
from .stable_diffusion_xl import ( from .stable_diffusion_xl import (
StableDiffusionXLImg2ImgPipeline, StableDiffusionXLImg2ImgPipeline,
StableDiffusionXLInpaintPipeline, StableDiffusionXLInpaintPipeline,
StableDiffusionXLPipeline, StableDiffusionXLPipeline,
) )
from .wuerstchen import WuerstchenCombinedPipeline, WuerstchenDecoderPipeline from .wuerstchen import (
WuerstchenCombinedPipeline,
WuerstchenDecoderPipeline,
)
AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(