Spaces:
Runtime error
Runtime error
Squeeze un-edited latents
Browse files
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 |
|