mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-07 04:54:47 +08:00
Compare commits
1 Commits
single-fil
...
auto-pipel
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
90cca148de |
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user