| 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 |