File size: 410 Bytes
f499d3b
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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)