Haoming02/sd-webui-old-photo-restoration
1
0
Fork 0
mirror of https://github.com/Haoming02/sd-webui-old-photo-restoration.git synced 2026-05-01 03:31:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix #19

Browse Source
This commit is contained in:
Haoming
2024-04-21 08:53:41 +08:00
committed by GitHub
parent 7686b038e5
commit e2df94c389
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Face_Enhancement/models/networks/__init__.py
View File

@@ -35,8 +35,7 @@ def modify_commandline_options(parser, is_train):
def create_network(cls, opt):
net = cls(opt)
net.print_network()
if len(opt.gpu_ids) > 0:
assert torch.cuda.is_available()
if torch.cuda.is_available() and len(opt.gpu_ids) > 0:
net.cuda()
net.init_weights(opt.init_type, opt.init_variance)
return net