diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-12 17:12:17 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-12 17:12:17 +0200 |
| commit | ef994e09872bcc3b4b34bd63d1e8302fff2bc0b2 (patch) | |
| tree | 950d7d45bd352ecdce887a2db7fafe4b105c9644 | |
| parent | 5811bcaf522a0a1c95861b4222c8a9a8da3ff358 (diff) | |
spaces... wtf
| -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())) |
