File size: 648 Bytes
80b58c8 28d98f8 80b58c8 28d98f8 80b58c8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | """
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",
]
|