Spaces:

rinong
/

StyleGAN-NADA

Runtime error

rinong commited on May 16, 2022

Commit

ea00796

•

1 Parent(s): 5479d05

Added explicit device to e4e construction

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ class ImageEditor(object):
         opts["checkpoint_path"] = model_path
         opts = Namespace(**opts)
-        self.e4e_net = pSp(opts)
         self.e4e_net.eval()
         self.e4e_net.cuda()

         opts["checkpoint_path"] = model_path
         opts = Namespace(**opts)
+        self.e4e_net = pSp(opts, self.device)
         self.e4e_net.eval()
         self.e4e_net.cuda()