Commit 26f733a0 authored by Elias Oenal's avatar Elias Oenal Committed by AUTOMATIC1111

fix for codeformer switching torch devices on metal systems.

parent bb2732c1
...@@ -53,7 +53,7 @@ def setup_codeformer(): ...@@ -53,7 +53,7 @@ def setup_codeformer():
def create_models(self): def create_models(self):
if self.net is not None and self.face_helper is not None: 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 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) 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(): ...@@ -66,7 +66,7 @@ def setup_codeformer():
self.net = net self.net = net
self.face_helper = face_helper self.face_helper = face_helper
self.net.to(shared.device) self.net.to(shared.device_codeformer)
return net, face_helper return net, face_helper
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment