Logic
16 lines
660 B
Python
16 lines
660 B
Python
# Backbone models
|
|
__all__ = ["LEWMViTBackbone", "ResNetBackbone", "ResNetDiffusionBackbone", "SigLIP2DiffusionBackbone"]
|
|
|
|
|
|
def __getattr__(name):
|
|
if name == "LEWMViTBackbone":
|
|
from .lewm_vit_backbone import LEWMViTBackbone
|
|
return LEWMViTBackbone
|
|
if name == "SigLIP2DiffusionBackbone":
|
|
from .siglip2_diffusion_backbone import SigLIP2DiffusionBackbone
|
|
return SigLIP2DiffusionBackbone
|
|
if name in {"ResNetBackbone", "ResNetDiffusionBackbone"}:
|
|
from .resnet_diffusion import ResNetDiffusionBackbone
|
|
return ResNetDiffusionBackbone
|
|
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|