Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
0f06396
1
Parent(s):
aba3fef
Update e4e_projection.py
Browse files- e4e_projection.py +1 -0
e4e_projection.py
CHANGED
|
@@ -8,6 +8,7 @@ from argparse import Namespace
|
|
| 8 |
from e4e.models.psp import pSp
|
| 9 |
from util import *
|
| 10 |
|
|
|
|
| 11 |
model_path = 'models/e4e_ffhq_encode.pt'
|
| 12 |
ckpt = torch.load(model_path, map_location='cpu')
|
| 13 |
opts = ckpt['opts']
|
|
|
|
| 8 |
from e4e.models.psp import pSp
|
| 9 |
from util import *
|
| 10 |
|
| 11 |
+
device = 'cpu'
|
| 12 |
model_path = 'models/e4e_ffhq_encode.pt'
|
| 13 |
ckpt = torch.load(model_path, map_location='cpu')
|
| 14 |
opts = ckpt['opts']
|