DeepFaceLab/models/__init__.py
2018-12-24 13:45:40 +04:00

8 lines
No EOL
289 B
Python

from .ModelBase import ModelBase
from .ConverterBase import ConverterBase
from .ConverterMasked import ConverterMasked
from .ConverterImage import ConverterImage
def import_model(name):
module = __import__('Model_'+name, globals(), locals(), [], 1)
return getattr(module, 'Model')