Spaces:
Runtime error
Runtime error
Added explicit device to e4e construction
Browse files
app.py
CHANGED
@@ -90,7 +90,7 @@ class ImageEditor(object):
|
|
90 |
opts["checkpoint_path"] = model_path
|
91 |
opts = Namespace(**opts)
|
92 |
|
93 |
-
self.e4e_net = pSp(opts)
|
94 |
self.e4e_net.eval()
|
95 |
self.e4e_net.cuda()
|
96 |
|
|
|
90 |
opts["checkpoint_path"] = model_path
|
91 |
opts = Namespace(**opts)
|
92 |
|
93 |
+
self.e4e_net = pSp(opts, self.device)
|
94 |
self.e4e_net.eval()
|
95 |
self.e4e_net.cuda()
|
96 |
|