|
from dataclasses import dataclass |
|
from typing import List, Optional, Union |
|
|
|
import numpy as np |
|
import PIL |
|
from PIL import Image |
|
|
|
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 ShapEPipeline |
|
else: |
|
from .blip_image_processing import BlipImageProcessor |
|
from .modeling_blip2 import Blip2QFormerModel |
|
from .modeling_ctx_clip import ContextCLIPTextModel |
|
from .pipeline_blip_diffusion import BlipDiffusionPipeline |
|
|