victorisgeek's picture
Upload 3 files
965bd28 verified
raw
history blame contribute delete
267 Bytes
from .inswapper import InSwapper
def get_swapper_model(name='', root=None, **kwargs):
if name.lower() == 'inswapper':
return InSwapper(name=name, root=root, **kwargs)
else:
raise UserWarning('The swapper model {} not support.'.format(name))