Save resulting latent

After editing, when saving the image, you also save the resulting latent for continuing working on it later on (or e.g. generate interpolations)
PeacefulHorizon · Jun 26, 2023 · b76c49a · b76c49a
1 parent 9b7e474
commit b76c49a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/viz/renderer.py b/viz/renderer.py
@@ -382,5 +382,6 @@ def _render_drag_impl(self, res,
             img = img.cpu().numpy()
             img = Image.fromarray(img)
         res.image = img
+        res.w = ws.cpu().numpy()
 
 #----------------------------------------------------------------------------