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 |
|