LOADERS = {} ## 用来注册 def register_loader(name): def register_loader_cls(cls): if name in LOADERS: return LOADERS[name] LOADERS[name] = cls return cls return register_loader_cls try: from .tbstars_ecom_emb import TBStars_ecom_emb_ModelLoader except ImportError as e: import warnings warnings.warn(f"tbstars_ecom_emb loader unavailable (skipped): {e}") ## 增加proj的loader from .proj_models import load_mlp_projector, load_add_projector