Upload bytedream/__init__.py with huggingface_hub
Browse files- bytedream/__init__.py +3 -0
bytedream/__init__.py
CHANGED
|
@@ -10,6 +10,7 @@ from .generator import ByteDreamGenerator
|
|
| 10 |
from .model import UNet2DConditionModel, AutoencoderKL, CLIPTextModel
|
| 11 |
from .pipeline import ByteDreamPipeline
|
| 12 |
from .scheduler import DDIMScheduler
|
|
|
|
| 13 |
|
| 14 |
__all__ = [
|
| 15 |
"ByteDreamGenerator",
|
|
@@ -18,4 +19,6 @@ __all__ = [
|
|
| 18 |
"CLIPTextModel",
|
| 19 |
"ByteDreamPipeline",
|
| 20 |
"DDIMScheduler",
|
|
|
|
|
|
|
| 21 |
]
|
|
|
|
| 10 |
from .model import UNet2DConditionModel, AutoencoderKL, CLIPTextModel
|
| 11 |
from .pipeline import ByteDreamPipeline
|
| 12 |
from .scheduler import DDIMScheduler
|
| 13 |
+
from .hf_api import HuggingFaceAPI, ByteDreamHFClient
|
| 14 |
|
| 15 |
__all__ = [
|
| 16 |
"ByteDreamGenerator",
|
|
|
|
| 19 |
"CLIPTextModel",
|
| 20 |
"ByteDreamPipeline",
|
| 21 |
"DDIMScheduler",
|
| 22 |
+
"HuggingFaceAPI",
|
| 23 |
+
"ByteDreamHFClient",
|
| 24 |
]
|