Skip to content

Commit 98079c2

Browse files
authored
Revert "Minimize re-exports from __init__ files (#44)"
This reverts commit 57862fb.
1 parent 57862fb commit 98079c2

File tree

5 files changed

+14
-7
lines changed

5 files changed

+14
-7
lines changed

sgm/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,5 @@
1+
from .data import StableDataModuleFromConfig
2+
from .models import AutoencodingEngine, DiffusionEngine
3+
from .util import instantiate_from_config, get_configs_path
4+
15
__version__ = "0.0.1"

sgm/data/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from .dataset import StableDataModuleFromConfig

sgm/models/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
from .autoencoder import AutoencodingEngine
2+
from .diffusion import DiffusionEngine

sgm/modules/__init__.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,5 @@
11
from .encoders.modules import GeneralConditioner
22

3-
__all__ = [
4-
# `sgm.models.GeneralConditioner` is referenced in model configurations, etc.,
5-
# so it must be re-exported from this module.
6-
"GeneralConditioner",
7-
"UNCONDITIONAL_CONFIG",
8-
]
9-
103
UNCONDITIONAL_CONFIG = {
114
"target": "sgm.modules.GeneralConditioner",
125
"params": {"emb_models": []},
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
from .denoiser import Denoiser
2+
from .discretizer import Discretization
3+
from .loss import StandardDiffusionLoss
4+
from .model import Model, Encoder, Decoder
5+
from .openaimodel import UNetModel
6+
from .sampling import BaseDiffusionSampler
7+
from .wrappers import OpenAIWrapper

0 commit comments

Comments
 (0)