rinong commited on
Commit
cf27c3e
1 Parent(s): 16c6665

Squeeze un-edited latents

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -208,7 +208,7 @@ class ImageEditor(object):
208
 
209
  # if edit type is none or if all sliders were set to 0
210
  if not target_latents:
211
- target_latents = [source_latent, ] * max((len(generators) - 1), 1)
212
 
213
  return target_latents
214
 
 
208
 
209
  # if edit type is none or if all sliders were set to 0
210
  if not target_latents:
211
+ target_latents = [source_latent.squeeze(0), ] * max((len(generators) - 1), 1)
212
 
213
  return target_latents
214