aaaaaaaaaaaaaaaaa
This commit is contained in:
parent
89d94e13a7
commit
834b6e396b
1 changed files with 1 additions and 1 deletions
2
webui.py
2
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.opts_channelslast:
|
||||
if cmd_opts.opt_channelslast:
|
||||
model = model.to(memory_format=torch.channels_last)
|
||||
model.eval()
|
||||
return model
|
||||
|
|
Loading…
Reference in a new issue