Enzo8930302 commited on
Commit
28d98f8
·
verified ·
1 Parent(s): 0e3999b

Upload bytedream/__init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
  ]