diff options
| -rwxr-xr-x | models/models.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/models/models.py b/models/models.py index 8e72e46..f5bc8d0 100755 --- a/models/models.py +++ b/models/models.py @@ -4,14 +4,14 @@ import torch def create_model(opt): if opt.model == 'pix2pixHD': - from .pix2pixHD_model import Pix2PixHDModel, InferenceModel - if opt.isTrain: + from .pix2pixHD_model import Pix2PixHDModel, InferenceModel + if opt.isTrain: model = Pix2PixHDModel() else: model = InferenceModel() else: - from .ui_model import UIModel - model = UIModel() + from .ui_model import UIModel + model = UIModel() model.initialize(opt) if opt.verbose: print("model [%s] was created" % (model.name())) |