From 26f733a026ecdd0449978b08fed1f024cb69ab65 Mon Sep 17 00:00:00 2001 From: Elias Oenal Date: Wed, 14 Sep 2022 13:44:00 +0200 Subject: [PATCH] fix for codeformer switching torch devices on metal systems. --- modules/codeformer_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/codeformer_model.py b/modules/codeformer_model.py index 21c704f7b..336f007d6 100644 --- a/modules/codeformer_model.py +++ b/modules/codeformer_model.py @@ -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