| from .unirig_ar import UniRigAR | |
| from .unirig_skin import UniRigSkin | |
| from .spec import ModelSpec | |
| def get_model(**kwargs) -> ModelSpec: | |
| MAP = { | |
| 'unirig_ar': UniRigAR, | |
| 'unirig_skin': UniRigSkin, | |
| } | |
| __target__ = kwargs['__target__'] | |
| del kwargs['__target__'] | |
| assert __target__ in MAP, f"expect: [{','.join(MAP.keys())}], found: {__target__}" | |
| return MAP[__target__](**kwargs) |