24 lines
852 B
Python
24 lines
852 B
Python
from ...utils import (
|
|
OptionalDependencyNotAvailable,
|
|
is_torch_available,
|
|
is_transformers_available,
|
|
)
|
|
|
|
|
|
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_kandinsky import KandinskyPipeline
|
|
from .pipeline_kandinsky_combined import (
|
|
KandinskyCombinedPipeline,
|
|
KandinskyImg2ImgCombinedPipeline,
|
|
KandinskyInpaintCombinedPipeline,
|
|
)
|
|
from .pipeline_kandinsky_img2img import KandinskyImg2ImgPipeline
|
|
from .pipeline_kandinsky_inpaint import KandinskyInpaintPipeline
|
|
from .pipeline_kandinsky_prior import KandinskyPriorPipeline, KandinskyPriorPipelineOutput
|
|
from .text_encoder import MultilingualCLIP
|