DeepFaceLab/models/__init__.py
2020-01-21 18:43:39 +04:00

6 lines
185 B
Python

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