Files
generative-models/sgm/modules/__init__.py
Aarni Koskela 57862fb4c7 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).
2023-07-25 16:24:09 +02:00

14 lines
361 B
Python

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": []},
}