/
ai-forever
/
diffusers-new
Обзор
Документация
Войти
/
ai-forever
/
diffusers-new
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
main
src/diffusers/pipelines/blip_diffusion/__init__.py
20 строк
697 B
Ayush Mangal
Add BLIP Diffusion (#4388)
21 сен 2023, 19:05
Не верифицирован
21 сен 2023, 19:05
157c901
Код
Авторство
О чём код?
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