Skip to content

Commit 45c06c4

Browse files
authored
Merge pull request #86 from nivha/support_multiple_file_projections
fixed to allow multiple projections
2 parents 0ecf8ff + ae2b5e8 commit 45c06c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎projector.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def make_image(tensor):
130130

131131
noises = g_ema.make_noise()
132132

133-
latent_in = latent_mean.detach().clone().unsqueeze(0)
133+
latent_in = latent_mean.detach().clone().unsqueeze(0).repeat(imgs.shape[0], 1)
134134

135135
if args.w_plus:
136136
latent_in = latent_in.unsqueeze(1).repeat(1, g_ema.n_latent, 1)

0 commit comments

Comments
 (0)