convert VAE moments back to np

2023-04-29 16:02:09 -05:00 · 2023-04-29 16:02:09 -05:00 · 47c7f5d1da
parent b1a989b71c
commit 47c7f5d1da
1 changed files with 1 additions and 1 deletions
--- a/api/onnx_web/diffusers/patches/vae.py
+++ b/api/onnx_web/diffusers/patches/vae.py
@ -122,7 +122,7 @@ class VAEWrapper(object):
                result_row.append(tile[:, :, :row_limit, :row_limit])
            result_rows.append(torch.cat(result_row, dim=3))

-        moments = torch.cat(result_rows, dim=2)
+        moments = torch.cat(result_rows, dim=2).numpy()
        if not return_dict:
            return (moments,)