From 89d94e13a72eea8c1d32d1c13b399685cb21fc6b Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Mon, 12 Sep 2022 16:51:23 +0300 Subject: [PATCH] rename --channelslast to --opt-channelslast to be in line with other torch optimizations --- webui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/webui.py b/webui.py index cc666e6f..d8d64be8 100644 --- a/webui.py +++ b/webui.py @@ -49,7 +49,7 @@ def load_model_from_config(config, ckpt, verbose=False): if len(u) > 0 and verbose: print("unexpected keys:") print(u) - if cmd_opts.channelslast: + if cmd_opts.opts_channelslast: model = model.to(memory_format=torch.channels_last) model.eval() return model