michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix for codeformer switching torch devices on metal systems.

Browse source
This commit is contained in:
Elias Oenal 2022-09-14 13:44:00 +02:00 committed by AUTOMATIC1111
parent bb2732c1c7
commit 26f733a026
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/codeformer_model.py
View file

@ -53,7 +53,7 @@ def setup_codeformer():
def create_models(self):
if self.net is not None and self.face_helper is not None:
self.net.to(shared.device)
self.net.to(shared.device_codeformer)
return self.net, self.face_helper
net = net_class(dim_embd=512, codebook_size=1024, n_head=8, n_layers=9, connect_list=['32', '64', '128', '256']).to(devices.device_codeformer)
@ -66,7 +66,7 @@ def setup_codeformer():
self.net = net
self.face_helper = face_helper
self.net.to(shared.device)
self.net.to(shared.device_codeformer)
return net, face_helper