ByteDream / bytedream /__init__.py
Enzo8930302's picture
Upload bytedream/__init__.py with huggingface_hub
28d98f8 verified
raw
history blame contribute delete
648 Bytes
"""
Byte Dream - AI Image Generation Model
Production-ready text-to-image diffusion model optimized for CPU inference
"""
__version__ = "1.0.0"
__author__ = "Byte Dream Team"
from .generator import ByteDreamGenerator
from .model import UNet2DConditionModel, AutoencoderKL, CLIPTextModel
from .pipeline import ByteDreamPipeline
from .scheduler import DDIMScheduler
from .hf_api import HuggingFaceAPI, ByteDreamHFClient
__all__ = [
"ByteDreamGenerator",
"UNet2DConditionModel",
"AutoencoderKL",
"CLIPTextModel",
"ByteDreamPipeline",
"DDIMScheduler",
"HuggingFaceAPI",
"ByteDreamHFClient",
]