adirik commited on
Commit
88c803f
1 Parent(s): 74880fc

update app

Browse files
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
encoder4editing/.DS_Store ADDED
Binary file (6.15 kB). View file
 
encoder4editing/models/.DS_Store ADDED
Binary file (6.15 kB). View file
 
encoder4editing/models/stylegan2/.DS_Store ADDED
Binary file (6.15 kB). View file
 
psp_wrapper.py CHANGED
@@ -53,5 +53,5 @@ class psp_encoder:
53
 
54
 
55
  with torch.no_grad():
56
- _, latents = self.net(transformed_image.unsqueeze(0).to(device).float(), randomize_noise=False, return_latents=True)
57
  return latents.cpu().numpy()
 
53
 
54
 
55
  with torch.no_grad():
56
+ _, latents = self.net(transformed_image.unsqueeze(0).to(self.device).float(), randomize_noise=False, return_latents=True)
57
  return latents.cpu().numpy()