Spaces:

akhaliq
/

JoJoGAN

Runtime error

akhaliq HF Staff commited on Feb 17, 2022

Commit

f7d82f4

1 Parent(s): e4946b6

Update e4e_projection.py

Files changed (1) hide show

e4e_projection.py CHANGED Viewed

@@ -14,7 +14,7 @@ from util import *
 def projection(img, name, device='cuda'):
-    model_path = 'models/e4e_ffhq_encode.pt'
     ckpt = torch.load(model_path, map_location='cpu')
     opts = ckpt['opts']
     opts['checkpoint_path'] = model_path

 def projection(img, name, device='cuda'):
+    model_path = 'e4e_ffhq_encode.pt'
     ckpt = torch.load(model_path, map_location='cpu')
     opts = ckpt['opts']
     opts['checkpoint_path'] = model_path