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,7 +17,20 @@ import inspect
from collections import OrderedDict
from ..configuration_utils import ConfigMixin
from ..utils import DIFFUSERS_CACHE
from ..utils 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,
StableDiffusionControlNetInpaintPipeline,
@@ -25,7 +38,11 @@ from .controlnet import (
StableDiffusionXLControlNetImg2ImgPipeline,
StableDiffusionXLControlNetPipeline,
)
from .deepfloyd_if import IFImg2ImgPipeline, IFInpaintingPipeline, IFPipeline
from .deepfloyd_if import (
IFImg2ImgPipeline,
IFInpaintingPipeline,
IFPipeline,
)
from .kandinsky import (
KandinskyCombinedPipeline,
KandinskyImg2ImgCombinedPipeline,
@@ -52,7 +69,10 @@ from .stable_diffusion_xl import (
StableDiffusionXLInpaintPipeline,
StableDiffusionXLPipeline,
)
from .wuerstchen import WuerstchenCombinedPipeline, WuerstchenDecoderPipeline
from .wuerstchen import (
WuerstchenCombinedPipeline,
WuerstchenDecoderPipeline,
)
AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(