akhaliq HF staff commited on
Commit
f7d82f4
1 Parent(s): e4946b6

Update e4e_projection.py

Browse files
Files changed (1) hide show
  1. e4e_projection.py +1 -1
e4e_projection.py CHANGED
@@ -14,7 +14,7 @@ from util import *
14
  def projection(img, name, device='cuda'):
15
 
16
 
17
- model_path = 'models/e4e_ffhq_encode.pt'
18
  ckpt = torch.load(model_path, map_location='cpu')
19
  opts = ckpt['opts']
20
  opts['checkpoint_path'] = model_path
 
14
  def projection(img, name, device='cuda'):
15
 
16
 
17
+ model_path = 'e4e_ffhq_encode.pt'
18
  ckpt = torch.load(model_path, map_location='cpu')
19
  opts = ckpt['opts']
20
  opts['checkpoint_path'] = model_path