Revert "Minimize re-exports from __init__ files (#44)" (#63)

This reverts commit 57862fb4c7.
This commit is contained in:
Tim Dockhorn
2023-07-26 04:26:28 -04:00
committed by GitHub
parent 1da250906d
commit 7934245835
5 changed files with 14 additions and 7 deletions

View File

@@ -1 +1,5 @@
from .data import StableDataModuleFromConfig
from .models import AutoencodingEngine, DiffusionEngine
from .util import instantiate_from_config, get_configs_path
__version__ = "0.0.1"

View File

@@ -0,0 +1 @@
from .dataset import StableDataModuleFromConfig

View File

@@ -0,0 +1,2 @@
from .autoencoder import AutoencodingEngine
from .diffusion import DiffusionEngine

View File

@@ -1,12 +1,5 @@
from .encoders.modules import GeneralConditioner
__all__ = [
# `sgm.models.GeneralConditioner` is referenced in model configurations, etc.,
# so it must be re-exported from this module.
"GeneralConditioner",
"UNCONDITIONAL_CONFIG",
]
UNCONDITIONAL_CONFIG = {
"target": "sgm.modules.GeneralConditioner",
"params": {"emb_models": []},

View File

@@ -0,0 +1,7 @@
from .denoiser import Denoiser
from .discretizer import Discretization
from .loss import StandardDiffusionLoss
from .model import Model, Encoder, Decoder
from .openaimodel import UNetModel
from .sampling import BaseDiffusionSampler
from .wrappers import OpenAIWrapper