Spaces:
Running
on
Zero
Running
on
Zero
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) |