File size: 502 Bytes
0ab2566 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | 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 |