Fix device handling in FSEFull and FSEInverter: convert device string to torch.device object

models/methods.py

@@ -38,7 +38,7 @@ class FSEFull(nn.Module):
         self.opts.update(paths)
         self.opts = Namespace(**self.opts)
 
-        self.device = device
+        self.device = torch.device(device)
         self.inverter_pth = inverter_pth
 
         self.encoder = self.set_encoder()
@@ -169,7 +169,7 @@ class FSEInverter(nn.Module):
         self.opts.update(paths)
         self.opts = Namespace(**self.opts)
 
-        self.device = device
+        self.device = torch.device(device)
         self.encoder = self.set_encoder()
 
         self.decoder = Generator(self.opts.stylegan_size, 512, 8)