mirror of
https://github.com/Stability-AI/generative-models.git
synced 2025-12-24 08:44:20 +01:00
Minimize re-exports from __init__ files (#44)
This allows importing parts of the package without having to import practically everything (since importing a package will import its parents' __init__s, etc).
This commit is contained in:
@@ -1,5 +1 @@
|
||||
from .data import StableDataModuleFromConfig
|
||||
from .models import AutoencodingEngine, DiffusionEngine
|
||||
from .util import instantiate_from_config, get_configs_path
|
||||
|
||||
__version__ = "0.0.1"
|
||||
|
||||
@@ -1 +0,0 @@
|
||||
from .dataset import StableDataModuleFromConfig
|
||||
|
||||
@@ -1,2 +0,0 @@
|
||||
from .autoencoder import AutoencodingEngine
|
||||
from .diffusion import DiffusionEngine
|
||||
|
||||
@@ -1,5 +1,12 @@
|
||||
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": []},
|
||||
|
||||
@@ -1,7 +0,0 @@
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user